添加状态分类和液面分割

This commit is contained in:
琉璃月光
2025-09-01 14:14:18 +08:00
parent 6e553f6a20
commit ad52ab9125
2379 changed files with 102501 additions and 1465 deletions

View File

@ -152,12 +152,12 @@ def labelme_to_yolo_keypoints_batch(json_dir, output_dir, target_box_label="J1",
# ================== 用户配置区 ==================
JSON_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/3/folder_end"
OUTPUT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/3/labels_keypoints"
JSON_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/yolodataset/point1/f11"
OUTPUT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/yolodataset/point1/labels_keypoints"
TARGET_BOX_LABEL = "J1"
CLASS_ID = 0
CLASS_ID = 1
IMG_SHAPE = (1440, 2506) # (height, width)
KEYPOINTS_PER_INSTANCE = 4
KEYPOINTS_PER_INSTANCE = 3
# ================== 执行转换 ==================
if __name__ == "__main__":