Scaled yolov4-tiny, problems with running in pytorch.

WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network

GNU General Public License v3.0

2.01k stars 570 forks source link

Scaled yolov4-tiny, problems with running in pytorch. #178

Open Duarte-Nunes opened 3 years ago

Duarte-Nunes commented 3 years ago

Hi, i'm currently trying to run the ScaledYOLOv4 version of yolov4-tiny, however both in the ScaledYOLOv4 repository and the PyTorch-YOLOv4 i run into an error "AssertionError: Unsupported fields ['group_id', 'resize'] in cfg/yolov4-tiny-scaled.cfg".

Am i doing something wrong or does this mean i should use the Yolov3 repository or even darknet?

WongKinYiu commented 3 years ago

from pytorch 1.6, the direct implementation of group_id will get error. i implemented it by other way, but the weights file can not be shared.

Duarte-Nunes commented 3 years ago

Thank you for your response, i can train yolo tiny now, i used your other yolov4 repository.