文件名称:VehicleColorID:使用YOLO执行对象分割,并使用颜色直方图和kNN分类器进行颜色识别
文件大小:53.35MB
文件格式:ZIP
更新时间:2024-03-10 10:23:25
Python
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分类器可预测检测到的对象的颜色。