VLM-Cholecystectomie / vit_labels.py
Bopa-Boptech's picture
Upload 5 files
433bf10 verified
PHASE_LABELS = {
"PREPARATION": 0,
"CALOT_TRIANGLE_DISSECTION": 1,
"CLIPPING_AND_CUTTING_CYSTIC_ELEMENTS": 2,
"GALLBLADDER_DISSECTION": 3,
"GALLBLADDER_PACKAGING": 4,
"CLEANING_AND_COAGULATION": 5,
"NOT_ANNOTATED": 6,
}
STEP_LABELS = {
"ADHESION_RELEASE": 0,
"EXPLORATION": 1,
"TROCAR_INTRODUCTION": 2,
"TROCAR_POSITION_DETERMINATION": 3,
"OUT": 4,
"POOR_VIEW": 5,
"EXPECTATION": 6,
"CALOT_TRIANGLE_EXPOSURE": 7,
"CALOT_TRIANGLE_OPENING": 8,
"SKELETONIZATION_OF_CYSTIC_ELEMENTS": 9,
"DISSECTION_SUB_HEPATIC_INFUNDIBULAR_PLANE": 10,
"ASPIRATION": 11,
"LIGATION_OF_CYSTIC_ELEMENTS": 12,
"SECTION_OF_CYSTIC_ELEMENTS": 13,
"STONE_RISE": 14,
"GALLBLADDER_TRACTION": 15,
"DISSECTION_AT_SUB_HEPATIC_PLANE": 16,
"BAG_MANIPULATION": 17,
"GALLBLADDER_MANIPULATION": 18,
"PLACEMENT_OF_GALLBLADDER_IN_THE_BAG": 19,
"CONTROL": 20,
"CLIP_CONTROL": 21,
"ANNOT_REMOVED": 22,
"DRAINAGE": 23,
"CLEANING": 24,
"EXPOSURE": 25,
"TROCAR_REMOVAL": 26,
"OPTIC_REMOVAL": 27,
"LIGATION_OF_BRANCH_OF_CYSTIC_ARTERY": 28,
"NOT_ANNOTATED": 29,
}
TARGET_LABELS = {
"ABDOMINAL_WALL_CAVITY": 0,
"ABDOMINAL_CAVITY": 1,
"LIVER": 2,
"PERITONEUM": 3,
"GALLBLADDER": 4,
"GUT": 5,
"DUODENUM": 6,
"STOMACH": 7,
"GREATER_OMENTUM": 8,
"HEPATO_DUODENAL_LIGAMENT": 9,
"NO_TARGET": 10,
"ADHESION": 11,
"ROUND_LIGAMENT": 12,
"GALLBLADDER_PERITONEUM": 13,
"CALOT_TRIANGLE_ELEMENTS": 14,
"CYSTIC_ARTERY": 15,
"CYSTIC_DUCT": 16,
"SUB_HEPATIC_INFUNDIBULAR_PLANE": 17,
"BRANCH_OF_THE_CYSTIC_ARTERY": 18,
"SEROSITY": 19,
"STONE": 20,
"SUB_HEPATIC_PLANE": 21,
"GALLBLADDER_BED": 22,
"CLIP": 23,
"BAG": 24,
"GALLBLADDER_BAG": 25,
"TROCAR": 26,
"NOT_ANNOTATED": 27,
"ANNOT_REMOVED": 28,
}