With coco dataset I can run the code. Now i want to use a custom dataset with just 3 classes?
What would be best approach to replace coco and integrate custom weights?

Any help would be appreciated- Thanks.