问题描述:
Traceback (most recent call last):
File "C:/Users/aoqia/Desktop/yolov7-pose-caoxie/labelme2yolo-points.py", line 19, in <module>
os.mkdir('labels')
FileExistsError: [WinError 183] 当文件已存在时,无法创建该文件。: 'labels'
Process finished with exit code 1
问题解决:我把待处理的文件夹名字改了一下,然后问题就解决了。刚开始是grpae111,改成了grape1111。
注意:也有可能是我把代码放在了其他地方运行解决的。
原因分析:网上看到的说代码鲁棒性弱会导致这个问题,大家可以尝试修改文件夹的名字或者代码的路径,来解决这个问题。
import os
import json
import shutil
import numpy as np
from tqdm import tqdm
Dataset_root = 'C:/Users/aoqia/Desktop/grape11111/labelme_jsons'
# 框的类别
bbox_class = {
'grape': 0
}
# 关键点的类别
keypoint_class = ['picking']
os.chdir(Dataset_root)
os.mkdir('labels')
os.mkdir('labels/train')
os.mkdir('labels/val')
os.mkdir('labels/test')
# 函数-处理单个labelme标注json文件
def process_single_json(labelme_path, save_folder='C:/Users/aoqia/Desktop/grape11111/labelme_jsons'):
with open(labelme_path, 'r', encoding='utf-8') as f:
labelme = json.load(f)
img_width = labelme['imageWidth'] # 图像宽度
img_height = labelme['imageHeight'] # 图像高度
# 生成 YOLO 格式的 txt 文件
suffix = labelme_path.split('.')[-2]
yolo_txt_path = suffix + '.txt'
with open(yolo_txt_path, 'w', encoding='utf-8') as f:
for each_ann in labelme['shapes']: # 遍历每个标注
if each_ann['shape_type'] == 'rectangle': # 每个框,在 txt 里写一行
yolo_str = ''
## 框的信息
# 框的类别 ID
bbox_class_id = bbox_class[each_ann['label']]
yolo_str += '{} '.format(bbox_class_id)
# 左上角和右下角的 XY 像素坐标
bbox_top_left_x = int(min(each_ann['points'][0][0], each_ann['points'][1][0]))
bbox_bottom_right_x = int(max(each_ann['points'][0][0], each_ann['points'][1][0]))
bbox_top_left_y = int(min(each_ann['points'][0][1], each_ann['points'][1][1]))
bbox_bottom_right_y = int(max(each_ann['points'][0][1], each_ann['points'][1][1]))
# 框中心点的 XY 像素坐标
bbox_center_x = int((bbox_top_left_x + bbox_bottom_right_x) / 2)
bbox_center_y = int((bbox_top_left_y + bbox_bottom_right_y) / 2)
# 框宽度
bbox_width = bbox_bottom_right_x - bbox_top_left_x
# 框高度
bbox_height = bbox_bottom_right_y - bbox_top_left_y
# 框中心点归一化坐标
bbox_center_x_norm = bbox_center_x / img_width
bbox_center_y_norm = bbox_center_y / img_height
# 框归一化宽度
bbox_width_norm = bbox_width / img_width
# 框归一化高度
bbox_height_norm = bbox_height / img_height
yolo_str += '{:.5f} {:.5f} {:.5f} {:.5f} '.format(bbox_center_x_norm, bbox_center_y_norm,
bbox_width_norm, bbox_height_norm)
## 找到该框中所有关键点,存在字典 bbox_keypoints_dict 中
bbox_keypoints_dict = {}
for each_ann in labelme['shapes']: # 遍历所有标注
if each_ann['shape_type'] == 'point': # 筛选出关键点标注
# 关键点XY坐标、类别
x = int(each_ann['points'][0][0])
y = int(each_ann['points'][0][1])
label = each_ann['label']
if (x > bbox_top_left_x) & (x < bbox_bottom_right_x) & (y < bbox_bottom_right_y) & (
y > bbox_top_left_y): # 筛选出在该个体框中的关键点
bbox_keypoints_dict[label] = [x, y]
## 把关键点按顺序排好
for each_class in keypoint_class: # 遍历每一类关键点
if each_class in bbox_keypoints_dict:
keypoint_x_norm = bbox_keypoints_dict[each_class][0] / img_width
keypoint_y_norm = bbox_keypoints_dict[each_class][1] / img_height
yolo_str += '{:.5f} {:.5f} {} '.format(keypoint_x_norm, keypoint_y_norm,
2) # 2-可见不遮挡 1-遮挡 0-没有点
else: # 不存在的点,一律为0
yolo_str += '0 0 0 '
# 写入 txt 文件中
f.write(yolo_str + '\n')
shutil.move(yolo_txt_path, save_folder)
print('{} --> {} 转换完成'.format(labelme_path, yolo_txt_path))
# 转换训练集标注文件至labels/train目录
os.chdir('C:/Users/aoqia/Desktop/grape11111/labelme_jsons/train')
save_folder = 'C:/Users/aoqia/Desktop/grape11111/labels/train'
for labelme_path in os.listdir():
process_single_json(labelme_path, save_folder=save_folder)
print('YOLO格式的txt标注文件已保存至 ', save_folder)
os.chdir('C:/Users/aoqia/Desktop/grape11111/labelme_jsons/val')
save_folder = 'C:/Users/aoqia/Desktop/grape11111/labels/val'
for labelme_path in os.listdir():
process_single_json(labelme_path, save_folder=save_folder)
print('YOLO格式的txt标注文件已保存至 ', save_folder)
os.chdir('C:/Users/aoqia/Desktop/grape11111/labelme_jsons/test')
save_folder = 'C:/Users/aoqia/Desktop/grape11111/labels/test'
for labelme_path in os.listdir():
process_single_json(labelme_path, save_folder=save_folder)
print('YOLO格式的txt标注文件已保存至 ', save_folder)