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 530 forks source link

isDifficult parameter check, default to 0 if not found in file #180

Open SanchithaD opened 2 years ago

SanchithaD commented 2 years ago

Some data doesn't have the is_difficult parameter. Created a check to look for is_difficult in file, when it's not there default the value to 0.