1. 环境配置
1.1 训练环境和onnx(电脑端执行)
#使用conda创建一个python环境
conda create -n torch python=3.9
#激活环境
conda activate torch
#安装yolov8
pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple
#onnx安装
pip i[......]
1. 环境配置
1.1 训练环境和onnx(电脑端执行)
#使用conda创建一个python环境
conda create -n torch python=3.9
#激活环境
conda activate torch
#安装yolov8
pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple
#onnx安装
pip i[......]
我的需求是只保留人(和没有人的副样本),抽样,其他忽略
import json
import os
import shutil
import random
def coco_to_yolo_bbox(image_width, image_height, bbox):
"""
Convert bbox from COCO format to YOLO format.
COCO format: [top_left_x, top_left_y, width,[......]
WIDER FACE是一个人脸数据集合,其中包含了人脸的标注,可以用它来转化为yolo的标注格式,做训练,极大节约成本
转化代码:
import os
def convert_wider_to_yolo(annotation_file, image_folder, output_folder):
if not os.path.exists(output_folder):
os.makedirs(output_folder)
with ope[......]