添加状态分类和液面分割
This commit is contained in:
@ -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__":
|
||||
|
||||
Reference in New Issue
Block a user