/US-Transporter-eval

Evaluation repository of Unsupervised Keypoint Detection models trained on Lung and Wrist Ultrasound videos

Primary LanguagePython

US-Transporter-eval

This is the evaluation repository of Keypoint models trained on Lung and Wrist Ultrasound relevant to the arXiv journal

Setup

git clone --recursive https://github.com/tripathiarpan20/US-Transporter-eval.git
cd US-Transporter-eval
pip install -r requirements.txt
mv phasepack/phasepack/* phasepack/

Model Checkpoints

Run the following commands to download LUS and WUS checkpoints respectively:

gdown --id 1HBuTOm_5-p7VrpYWYYPoJp2epQvbyPl9
gdown --id 1TMMNh6Vp07ejBDInsar5wXhljs9q7T57

Evaluation

  • Separate scripts for LUS & WUS checkpoints using pl_transporter.py to take input as JPEG, PNG etc images and output keypoint marked images in an output folder to be prepared
    • LUS
    • WUS

See US_Transporter_eval.ipynb for evaluation commands on Wrist and LUS Ultrasound images and videos, the outputs can be found in /content/US-Transporter-eval/output folder within the Colab session after executing relevant cells.

Transporter

References

  1. Third-party PyTorch Implementation of Transporter
  2. Phasepack
  3. PySSIM