Project

The proposed robotic arm would act as an additional hand to the doctor while performing any surgery by providing the requested surgical instrument on his/her request through speech command. The arm after processing these requests, would pick the instrument from the inventory tray which is prepared before the surgery and handover to the doctor.

Datset

The dataset comprised for 4 labels-

1. Scalpel
2. Curved Mayo scissors
3. Straight Mayo scissors
4. Straight discection clamp

Requirements

TensorFlow CPU

pip install -r requirements.txt

TensorFlow GPU

pip install -r requirements-gpu.txt

Weights File

Download the Weights

Save the weights into the 'data' folder of this repository and rename it as

custom.weights

Run below for creating the model

conda env create -f conda-cpu.yml
conda activate yolov4-cpu
python save_model.py --weights ./data/custom.weights --output ./checkpoints/custom-416 --input_s

To run the run the model

python detect_video.py --weights ./checkpoints/custom-416 --size 416 --model yolov4 --video 0 --output ./detections/results.avi

Contribution by:-

1. Aashay Patel
2. Anshul Rao
3. Aditya Ashok
4. Harivansh Narayan

Special thanks to The AI Guy for guidance.