jcfr / ci-sandbox

My sandbox for experimenting with CI services
Apache License 2.0
1 stars 4 forks source link

Project: Live tracked ultrasound processing with PyTorch #44

Open jcfr opened 1 year ago

jcfr commented 1 year ago

Category

IGT and Training

Key Investigators

Project Description

Our past code for training and deploying ultrasound segmentation in real time was based on TensorFlow. Example project: https://youtu.be/WyscpAee3vw

The goal for this project week is to provide a new open-source implementation using PyTorch and modern AI tools like MONAI and wandb. A Slicer module will also be provided to deploy trained AI on recorded or live ultrasound streams.

Objective

  1. Export annotated ultrasound+tracking data for training
  2. Example code for training
  3. Slicer module to use trained models on ultrasound data in Slicer

Approach and Plan

  1. All data processing and training code will be here: https://github.com/SlicerIGT/aigt/tree/master/UltrasoundSegmentation
  2. Slicer module will be here: https://github.com/SlicerIGT/aigt/tree/master/SlicerExtension/LiveUltrasoundAi/TorchLiveUs

Progress and Next Steps

  1. Describe specific steps you have actually done.

Illustrations

No response

Background and References

No response

github-actions[bot] commented 1 year ago

Project Page Pull Request Creation

:white_check_mark: COMPLETED: See https://github.com/jcfr/ci-sandbox/pull/45