RecON

This repository is the official implementation for "RecON: Online Learning for Sensorless Freehand 3D Ultrasound Reconstruction".

Environment

• PyTorch with GPU
• OpenCV-Python build from CUDA
• Run pip install -r requirements.txt

Training

• Backbone

  python3 -m main -m online_bk -d Spine -r hp_bk -g0
  

• Discriminator

  python3 -m main -m online_d -d Spine -r hp_d -g0
  

Online Learning

python3 -m main -m online_fm -d Spine -r hp_fm -g0 -t0

Demo

An interactive demo is available in here.