效果:

需要准备的材料:
1、首先需要将你的数据集转化为COCO指标的格式
2、用val.py保存json文件
3、将val的预测json和coco转换后的json进行比较,就能得到COCO指标
将数据集转换为coco指标格式
import os
import cv2
import json
from tqdm import tqdm
from sklearn.model_selection import train_test_split
import argparse
# visdrone2019 数据集的类别定义
classes = ['person','vest','blue helmet','red helmet','white helmet','yellow helmet']
# 命令行参数解析器
parser = argparse.ArgumentParser()
parser.add_argument('--image_path', default=r'D:\YOLO_up\data\datasets\CHV_dataset\images\val',type=str, help="图片路径")
parser.add_argument('--label_path', default=r'D:\YOLO_up\data\datasets\CHV_dataset\labels\val',type=str

订阅专栏 解锁全文
303

被折叠的 条评论
为什么被折叠?



