Open edbek opened 2 years ago
YOLOv5 and YOLOv4 are not compatible, since YOLOv5 is not YOLO-family descendant in terms of source ML framework - https://github.com/pjreddie/darknet/issues/2198.
You can use yolov7 (tiny version) though - https://github.com/AlexeyAB/darknet/issues/8595
Also this question could be addressed to OpenCV maintainers, since opencv::dnn is used.
A few more questions:
Thanks in advance for your replies !
vehicles-only
detector (for counting cars) https://github.com/LdDl/rust-road-traffic/tree/master/data (actual training steps are described at https://github.com/LdDl/yolo_vehicles ). What kind of data do you need find on your images? May be I've already seen such pretrained model.To count cars, it is necessary to add various types of trucks (maybe even with trailers), fixed-route taxis, trolleybuses, trams to the model. But there is only a limited set of vehicles in the existing model ("car", "motorbike", "bus", "train", "truck").
Thanks.
Yeah Pretty every model trained on COCO / BIT / AIC-HCMC datasets (or subsets). Those don't contain too many vehicle types. If you need to distinguish some classes you can manually annotate dataset, but it will lead increased inference time for neural network (more classes = more features = more job in convolutional layers).
I am new to this technology. Is it possible to use yolov5 instead of yolov4 ?