How to train this on custom dataset?

HaloTrouvaille / YOLO-Multi-Backbones-Attention

Model Compression—YOLOv3 with multi lightweight backbones(ShuffleNetV2 HuaWei GhostNet), attention, prune and quantization

493 stars 118 forks source link

Open ghost opened 3 years ago

ghost commented 3 years ago

How I can train it on a custom dataset? Where to change classes and input size?