M2cai16-tool-locations May 2026

# Draw boxes img_with_boxes = draw_bounding_boxes(img, boxes, labels=[class_names[l] for l in labels], colors='red', width=2) plt.figure(figsize=(10, 8)) plt.imshow(img_with_boxes.permute(1,2,0)) plt.axis('off') plt.title(f"Frame {idx} — {len(boxes)} tools detected") plt.show() dataset = M2CAI16ToolLocations('./m2cai16-tool-locations') show_annotations(dataset, idx=0) 4. Useful Preprocessing for Training Convert to COCO format (for Detectron2, MMDetection, etc.):

path: ./m2cai16-tool-locations train: images/train val: images/val nc: 16 names: ['grasper','scissors','hook','clipper','irrigator','specimen_bag','bipolar','hook_electrode','trocars','stapler','suction','clip_applier','vessel_sealer','ligasure','ultrasonic','other'] This guide gives you a production‑ready starting point for loading, visualizing, converting, and training on the dataset. Adjust class names and annotation JSON structure based on your exact dataset version. m2cai16-tool-locations

yolo detect train data=m2cai16.yaml model=yolov8n.pt epochs=100 imgsz=640 Example m2cai16.yaml : yolo detect train data=m2cai16

import matplotlib.pyplot as plt from torchvision.utils import draw_bounding_boxes from torchvision.transforms import ToTensor def show_annotations(dataset, idx=0): img, target = dataset[idx] if isinstance(img, torch.Tensor): img = (img * 255).byte() if img.max() <= 1 else img else: img = ToTensor()(img).byte() target = dataset[idx] if isinstance(img

def __init__(self, root_dir, transform=None): self.root_dir = root_dir self.transform = transform self.samples = [] # Collect all (frame_path, annotation_path) pairs ann_dir = os.path.join(root_dir, 'annotations') for ann_file in os.listdir(ann_dir): if not ann_file.endswith('.json'): continue ann_path = os.path.join(ann_dir, ann_file) video_id = ann_file.replace('.json', '') frame_dir = os.path.join(root_dir, 'frames', video_id) with open(ann_path, 'r') as f: annotations = json.load(f) for frame_name, boxes_info in annotations.items(): frame_path = os.path.join(frame_dir, frame_name) if os.path.exists(frame_path): self.samples.append((frame_path, boxes_info))