annotate / config.json
aiweld's picture
Create config.json
9f72551 verified
{
"architectures": ["YOLOv8Seg"],
"framework": "pytorch",
"model_type": "ultralytics-yolov8-seg",
"pipeline_tag": "image-segmentation",
"id2label": {
"0": "Porosity",
"1": "Slag Inclusion",
"2": "Incomplete Penetration",
"3": "Incomplete Fusion",
"4": "Wormhole",
"5": "Crack",
"6": "IQI",
"7": "Norm",
"8": "Linear Porosity",
"9": "Artifact",
"10": "Weld"
},
"label2id": {
"Porosity": 0,
"Slag Inclusion": 1,
"Incomplete Penetration": 2,
"Incomplete Fusion": 3,
"Wormhole": 4,
"Crack": 5,
"IQI": 6,
"Norm": 7,
"Linear Porosity": 8,
"Artifact": 9,
"Weld": 10
}
}