Closed OleksiiYeromenko closed 3 years ago
Ok, mistake in classy_track.py
- line 145 - missing stride
:
dataset = LoadImages(source, img_size=imgsz)
dataset = LoadImages(source, img_size=imgsz, stride=stride)
Hi @OleksiiYeromenko , I'm glad you were able to find the fix. It looks like that argument was added in YOLOV5 version 5.0. This repo is currently based on YOLOV5 version 4.0
Hi, Thanks for cool work, but I'm receiving error
RuntimeError: Sizes of tensors must match except in dimension 1. Got 24 and 23 in dimension 2 (The offending index is 1)
when trying to run _6 models with image size 1280.