search

opendr-eu / opendr

A modular, open and non-proprietary toolkit for core robotic functionalities by harnessing deep learning
Apache License 2.0
611 stars 94 forks source link

Finetuned yolov5 for trucks #476

Closed BillMousta closed 8 months ago

BillMousta commented 8 months ago

This pull request introduces a new download method within the YOLOv5DetectorLearner class, enabling automated fetching of pre-trained models directly into the local environment. Additionally, it showcases an inference demo leveraging a YOLOv5s model fine-tuned for enhanced truck detection accuracy. This demonstration serves as a practical example of the model's improved performance in identifying trucks.

omichel commented 8 months ago

This PR has many conflicts. Could you please fix them before requesting a review?

passalis commented 8 months ago

I changed the target to develop, which I think resolved the conflicts. If tests pass, then I think we can proceed with the review.

passalis commented 8 months ago

If this PR is ready for review, please let us know.

tsampazk commented 8 months ago

Hmm it seems that you beat me to it. There are several stuff that need some polishing with this PR. I will implement the changes i commented on and open a new PR.