利用coco数据集训练yolov5

1.数据集准备

本文用COCO数据集来进行训练,我在这里使用了脚本,将coco格式的数据集转成了yolo的txt格式,即COCO转YOLO格式。

import os
import json
from tqdm import tqdm
import argparse
 
parser = argparse.ArgumentParser()
# 这里根据自己的json文件位置,换成自己的就行
parser.add_argument('--json_path',
                    default='xxx.json'

猜你喜欢

转载自blog.csdn.net/weixin_38346042/article/details/129741006