Team: Autobots
Team Members:
Abhinaba Bala (2020701001)
Neelabh Kumar (2020701003)
Ruchi Chauhan (2018711001)
Rupak Lazarus (2020701020)
Assigned TA: Sai Soorya Rao Veeravalli
MidEval Presentation File
Final Presentation File
This project is undertaken as a part of the Computer Vision coursework at IIIT Hyderabad in Spring semester 2021. The main inspiration of our work is from the very recent paper of TransUNet which merits both Transformers and U-Net, as a strong alternative for medical image segmentation. The overview and proposal can be found here.
Code structure:
-lists
-networks
CVmidEval.pdf
ProjectProposal.pdf
README.md
ScatterringCoeff_generator.m
US_generateMasks.py
Unet.ipynb
eval.py
rc_bcet.py
train.py
utils.py
Download the dataset,
For lung images: here
For ultrasound images: here
To generate the scattering-coefficients for augmenting in TranUnet, use ScatteringCoeff_generator.m
For processing the ultrasound images use the US_generateMasks.py file.
And, for processing the lung images use rc_bcet.py file.
python3 eval.py --dataset Ultrasound --batch_size 12 --model_path /ssd_scratch/cvit/rupraze/models/ultrasound/epoch_120.pth
python3 eval_scatcoeff.py.py --dataset Ultrasound --batch_size 12 --model_path /ssd_scratch/cvit/rupraze/models/scat_ultrasound/epoch_149.pth
To get the demo of the inference process, run the 'InferenceDemo.ipynb' file