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

Yolov5 trucks fixes #487

Closed tsampazk closed 8 months ago

tsampazk commented 8 months ago

Applied the fixes from the late review on #476.

@BillMousta

One general comment: I am not sure about the naming of the new feature. We use the word trucks which is not really accurate since the model is trained on tractors, so it could be misleading. Reading trucks i would intuitively expect pickup trucks or semis. Let me know if i misunderstand something regarding the naming. If you agree i can go ahead and rename everything in this PR.

BillMousta commented 8 months ago

Applied the fixes from the late review on #476.

@BillMousta

One general comment: I am not sure about the naming of the new feature. We use the word trucks which is not really accurate since the model is trained on tractors, so it could be misleading. Reading trucks i would intuitively expect pickup trucks or semis. Let me know if i misunderstand something regarding the naming. If you agree i can go ahead and rename everything in this PR.

The primary objective of this fine-tuned model was to enhance the vehicle detection capabilities, particularly in the agricultural sector. We operated under the assumption that trucks and tractors share similar visual characteristics. Therefore, during training, we categorized both trucks and tractors under the 'trucks' class. This approach resulted in a more robust model, compared to the alternative of introducing a separate 'tractors' class.