Train custom object detectio model

[dusty_nv]

@rsamvelyan

I didn’t find a tutorial on object detection though. Do you know where I can find one?

Hi @rsamvelyan, see this post: DIGITS or somthing else - #7 by dusty_nv

For PyTorch training of object detection, I’ve been preparing the tutorial. You can find the code here: GitHub - dusty-nv/pytorch-ssd: MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.