search

amirbar / DETReg

Official implementation of the CVPR 2022 paper "DETReg: Unsupervised Pretraining with Region Priors for Object Detection".
https://amirbar.net/detreg
Apache License 2.0
336 stars 46 forks source link

Hello, your work is great, I am very interested, how to do pre-training on a custom pre-training dataset #8

Closed CQIITLAB closed 3 years ago

CQIITLAB commented 3 years ago

Hello, your work is great, I am very interested, how to do pre-training on a custom pre-training dataset

mmaaz60 commented 3 years ago

Convert your dataset to COCO or VOC format and enjoy!

CQIITLAB commented 3 years ago

Convert your dataset to COCO or VOC format and enjoy!

I want to make a new pretraining dataset instead of ImageNet100. If you can give some advice, thank you very much

mmaaz60 commented 3 years ago

Convert your dataset to COCO or VOC format and enjoy!

I want to make a new pretraining dataset instead of ImageNet100. If you can give some advice, thank you very much

I am assuming that you already collected the images for making your new dataset. If that's the case, you may try converting your collected dataset to ImageNet format and follow the instructions in this repo for generating SS boxes and performing pretraining.