How to train the object detection models in this repository's model zoo?

onnx / models

A collection of pre-trained, state-of-the-art models in the ONNX format

http://onnx.ai/models/

Apache License 2.0

7.71k stars 1.38k forks source link

How to train the object detection models in this repository's model zoo? #209

Open monocongo opened 5 years ago

monocongo commented 5 years ago

I would like to train the YOLOv3 model from this repository's model zoo, but there is no documentation describing that process. Can anyone advise? Thanks in advance for suggestions or insight.

monocongo commented 4 years ago

To clarify, I'm interested in transfer learning, i.e. taking a pre-trained model and further training it using my custom dataset of images/annotations. I'd like to do this for the YOLO and SDD models included here.

ebarsoum commented 4 years ago

Currently, those models and ONNX support only inference. Regarding training and transfer learning, check the training work group: https://github.com/onnx/working-groups/tree/master/training