# 自动标注脚本 **Repository Path**: tommy0x7c00/automatic-annotation-script ## Basic Information - **Project Name**: 自动标注脚本 - **Description**: 输出 `YOLO` 格式的数据集 - **Primary Language**: Unknown - **License**: Not specified - **Default Branch**: master - **Homepage**: None - **GVP Project**: No ## Statistics - **Stars**: 0 - **Forks**: 0 - **Created**: 2024-07-12 - **Last Updated**: 2024-07-18 ## Categories & Tags **Categories**: Uncategorized **Tags**: None ## README # 自动标注脚本 - 输出 `YOLO` 格式的数据集 ## 环境 ```bash export PYTHONPATH=$PYTHONPATH:/workspace/auto-label ``` ## Start ### 自动标注 ```bash python3 -m auto_label.yolo -h ``` ```bash -h, --help show this help message and exit -i IMGS_DIR, --imgs_dir IMGS_DIR 原始图片文件夹地址 -o OUTPUT_DST, --output_dst OUTPUT_DST 数据集文件名称 -m MODEL, --model MODEL YOLO模型文件 -d DEVICE, --device DEVICE cuda index ``` ```bash python3 -m auto_label.yolo -m /workspace/auto-label/yolov8s_fangtev3.0.0.pt -i fangte_new_images_v4.0.0 -o fangtev4.0.2 -d 1 ``` ### 合并coco数据集 #### coco2yolo #### 更改coco类别 ```bash python3 -m auto_label.change_coco -h ``` ```bash python3 -m auto_label.change_coco \ -ii coco128_yolo/coco128/images/train2017 \ -io fangte_coco128_yolo/images \ -li coco128_yolo/coco128/labels/train2017 \ -lo fangte_coco128_yolo/labels \ -m '{"0": 6, "1": 0, "2": 7, "3": 5, "4": 6, "5": 8, "6": 10, "7": 10}' ```