qfgaohao / pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
https://medium.com/@smallfishbigsea/understand-ssd-and-implement-your-own-caa3232cd6ad
MIT License
1.39k stars 529 forks source link

Too much data for some classes in open image #157

Open shashwat623 opened 3 years ago

shashwat623 commented 3 years ago

Hey, Some classes in the open image data set have a lot of data, sometimes going up to a million. People working on colab cannot download that much. It would be good to add some feature in the open image downloader code which can allow us to set the limit.