YOLOv5在coco数据集的实战笔记

环境

项目链接为：https://github.com/ultralytics/yolov5.
操作系统：MacOS
IDE：Pycharm
python版本：anaconda Pyhon3.7
pytorch版本：torch 1.8.1

准备

• 下载数据集：https://aistudio.baidu.com/aistudio/datasetdetail/79834
  

• 新建一个conda环境：

conda create -n yolo5 python=3.7

• pycharm打开yolov5-master，安装依赖项：

pip install -r requirements.txt

• 新主目录新建如下文件夹：
• 将下载的train按你的比例拆分为：
  images/train：训练集
  images/val：测试集

修改id：

• 下载的数据中，category_id从1开始，而yolo5中classes为list，从0开始检索，
  因此需要把train.json中的categories下的id减1，并annotations字段中category_id减一
• 修改代码：

import json

def process_json(input_json_file, output_json_file):
    file_in = open(input_json_file, "r")
    file_out = open(output_json_file, "w")
    json_data = json.load(file_in)

    for i in json_data['annotations']:
         i['category_id'] -= 1

    # 将修改后的数据写回文件
    f