kindle icon indicating copy to clipboard operation
kindle copied to clipboard

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

Open aiTrainee opened this issue 3 years ago • 1 comments

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?

aiTrainee avatar Jan 19 '22 20:01 aiTrainee

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

JeiKeiLim avatar Jan 20 '22 02:01 JeiKeiLim