Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

yolov5 #22

Open
edbek opened this issue Aug 29, 2022 · 5 comments
Open

yolov5 #22

edbek opened this issue Aug 29, 2022 · 5 comments
Assignees
Labels
enhancement New feature or request help wanted Extra attention is needed

Comments

@edbek
Copy link

edbek commented Aug 29, 2022

I am new to this technology. Is it possible to use yolov5 instead of yolov4 ?

@edbek edbek added enhancement New feature or request help wanted Extra attention is needed labels Aug 29, 2022
@LdDl
Copy link
Owner

LdDl commented Aug 29, 2022

YOLOv5 and YOLOv4 are not compatible, since YOLOv5 is not YOLO-family descendant in terms of source ML framework - pjreddie/darknet#2198.

You can use yolov7 (tiny version) though - AlexeyAB/darknet#8595

Also this question could be addressed to OpenCV maintainers, since opencv::dnn is used.

@edbek
Copy link
Author

edbek commented Aug 30, 2022

A few more questions:

  1. What is faster yolov7 (tiny) or yolov4 (tiny)?
  2. Are there for yolov4, besides the standard set of labels ("coco.names"), any others with their already trained models? Or is there only one model for yolov4?

Thanks in advance for your replies !

@LdDl
Copy link
Owner

LdDl commented Aug 30, 2022

  1. tiny-v7 should be considered slower since there are more layers in network architecture.
  2. there are tons of pretrained models in internet. As example I have another repo with pretrained tiny-v4, tiny-v3 for 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.

@edbek
Copy link
Author

edbek commented Aug 31, 2022

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.

@LdDl
Copy link
Owner

LdDl commented Aug 31, 2022

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).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

2 participants