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.

References

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