GitHub - harshakmohan/robot_tool_segmentation: Semantic segmentation of surgical tools w/ UNET on UCL dVRK segmentation dataset

harshakmohan / robot_tool_segmentation Public

Notifications You must be signed in to change notification settings
Fork 0
Star 1

Semantic segmentation of surgical tools w/ UNET on UCL dVRK segmentation dataset

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 55 Commits
.gitignore		.gitignore
dataset.py		dataset.py
model.py		model.py
predict_from_checkpoint.py		predict_from_checkpoint.py
readme.txt		readme.txt
requirements.txt		requirements.txt
train.py		train.py
utils.py		utils.py

Repository files navigation

Developing a semantic segmentation classifier using the U-Net architecture to segment surgical tools in a robotic endoscopic image.

Dataset can be downloaded here: https://www.ucl.ac.uk/interventional-surgical-sciences/ex-vivo-dvrk-segmentation-dataset-kinematic-data