Open DWSuryo opened 3 years ago
On this sample, I'm using original YOLOv3-tiny from https://pjreddie.com/darknet/yolo/
Currently, I dont have any experience using YOLOv4-tiny. But on the error you are facing out, problem seems on utils.py script. That script is only the post-processing part after network finish generating output layer.
I guest, this because output layer from YOLOv3 and YOLOv4 is different.
utils.py on this test script only tested on YOLOv3 layer output , just check YOLOv4 layer output and than compare with YOLOv3
in your example, you used pretrained yolov3-tiny model for testing those sample images. i used pretrained yolov4-tiny models here (search yolov4-tiny in page) to see how it works. Specifications: Jetson Nano (with 4GB swap), Jetpack 4.5.1, OpenCV 4.5.2, numpy 1.19
Then I switched v4 and v3 including cfg files and vice-versa, looks like the problem might be in v4 weights. Have you experienced this yet? If I'm curious, is the v3 model pretrained by yours or using samples? Thank you