Creating yolov2-tiny.yaml to fit ultalytics/yolov3 code

JeiKeiLim / kindle

Making a PyTorch model easier than ever!

https://limjk.ai/kindle/

MIT License

78 stars 7 forks source link

Creating yolov2-tiny.yaml to fit ultalytics/yolov3 code #58

Open aiTrainee opened 2 years ago

aiTrainee commented 2 years ago

Hi, I'm trying to make a 'yolov2-tiny.yaml' similare to 'yolov3-tiny.yaml' to integrate it in this code: ultralytics/yolov3. tinyv2 I tried this, but got stuck in the head. Not sure that the backbone is alright too. Can you help me with that please?

JeiKeiLim commented 2 years ago

We use different class name for the head part which is YOLOHead. You can check this here https://limjk.ai/kindle/tutorial.html#5-make-object-detectiong-model-using-yolohead

Also, we have separate MaxPool class too. Although it might work almost same. I forgot if there is any difference between using nn.MaxPool2d and MaxPool :( You can check the reference here https://limjk.ai/kindle/modules.html#maxpool