VehicleColorID
使用YOLOv3进行对象分割,并使用颜色直方图和kNN分类器进行颜色识别。
需要图书馆
OpenCV-4.2
枕头-6.1
脾气暴躁-1.18.1
Matplotlib-3.1.3
其他需求
Darknet和YOLOv3 cfg和权重已安装。 按照上的说明安装Darknet。
color_feature_extractor和knn模块需要与VehicleColorID文件放置在同一文件夹中。
这些文件应放置要检测的颜色数据集。
训练数据生成
使用以下命令运行color_feature_extractor.py:python color_feature_extractor.py --path $ COLOR_DATASET_PATH $
运行分类器
在COCO数据集上预训练的YOLOv3用作对象检测器,基于颜色的kNN分类器可预测检测到的对象的颜色。
2022-03-11 15:23:00
53.35MB
Python
1