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

[Anonymous]

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




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



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