kindle
kindle copied to clipboard
Creating yolov2-tiny.yaml to fit ultalytics/yolov3 code
Hi,
I'm trying to make a 'yolov2-tiny.yaml' similare to 'yolov3-tiny.yaml' to integrate it in this code: ultralytics/yolov3.
I tried this, but got stuck in the head. Not sure that the backbone is alright too.
Can you help me with that please?
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