Skip to content

Developed a surgical video segmentation pipeline by using Detectron2, powered by GPT 3.5 for real-time inference

License

Notifications You must be signed in to change notification settings

Patil-Ojas/Surgical-Tool-Detector

Repository files navigation

Stryker Hackathon

⚙️ Problem Statement

Tasked with developing deep learning models that can perform the task of segmenting and recognizing the instruments shown in the videos. In each frame, a variety of surgical tools are present (Instruments, Clamps, threads). These tools need to be segmented according to their category along with the tissue appearing in the background.

🕵️ Surgical-Tool-Detector

-> Built a surgical video segmentation pipeline by fine-tuning Detectron2, a modular framework by Meta AI Research, and Mask-RCNN for binary classification and surgical tool recognition.

-> Developed Pipeline for automated conversion of segmented video frames to quantifiable files and used GPT-3.5 for real-time inferencing.

📺 Demo

Live Demo

⏳ Workflow

image

📂 Dataset

The training dataset is made up of 16 robotic procedures. The original video data was recorded at 60 Hz and to reduce labelling cost they have been subsampled to 2 Hz. Sequences with little or no motion are manually removed to leave 149 frames per procedure. Video frames are 1280x1024 for which labeling is provided.

The classes found in the training and test are:

  • Instrument
  • Drop in Ultrasound Probe
  • Suturing Needles
  • Suturing thread
  • Clips/clamps
  • Background tissue

Each class will have a distinct numerical label in a ground truth image. A supplied json file will contain the class name to numerical label mapping.

Inferencing using GPT-3.5

image image

🤝 Team Xerneas

  • Ojas Patil
  • Garvita Kesarwani

🖋️ License

Achape-License-2

About

Developed a surgical video segmentation pipeline by using Detectron2, powered by GPT 3.5 for real-time inference

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published