Модель Yolov5, захватывающая другие объекты, кроме предполагаемого или желаемого объекта. ⇐ Python

[Anonymous]

Вопрос
У меня есть модель YOLO V5 m для набора данных для обнаружения определенного продукта, но после обучения набора данных, состоящего из различных изображений для продукта в разных случаях, сценариях и различных условия освещения без повторения случаев, на 50 эпохах, средней архитектуре Yolo v5 и размере пакета 16. Я заметил, что модель зависит от цветовых характеристик, а это означает, что обнаруживаемые объекты имеют желтый цвет. и его оттенки. Я использовал train.py, который поставляется с клоном yolo на github. Что мне следует сделать, чтобы решить эту проблему? Пожалуйста, помогите мне, а также найдите ниже используемый набор данных для обучения (около 450 изображений).
Продукт, который необходимо обнаружить:




Другие обнаруженные продукты:


Набор данных использовано в обучении (около 450 изображений) на Google Диске:
https://drive.google.com/file/d/1oBH28f ... sp=sharing

Подробнее здесь: https://stackoverflow.com/questions/792 ... red-object