search

deeplearning-wisc / stud

source code for CVPR'22 paper "Unknown-Aware Object Detection: Learning What You Don’t Know from Videos in the Wild"
Apache License 2.0
117 stars 14 forks source link

Custom Dataset #10

Closed hasnainali659 closed 1 year ago

hasnainali659 commented 1 year ago

How can we modify the code base for our custom dataset?

d12306 commented 1 year ago

hi, @hasnainali659 , thanks for the interest in our work! for custom datasets, you should convert it into the video-based coco format, as we did in this file. https://github.com/deeplearning-wisc/stud/blob/main/datasets/bdd100k2coco.py

hasnainali659 commented 1 year ago

Thanks for the response. After this step for training do we have to modify srce/builtin.py file also?

d12306 commented 1 year ago

@hasnainali659 , yes! you can then register the dataset for training and inference.