An attempt at tumor segmentation with UNET and SegNet on the lung tumor dataset from the Medical Decathlon data.
MIT License
15
stars
3
forks
source link
hi, when using the inference.py and loading the ckpt file you have provided, I faced the problem "Can't get attribute 'SegNet' on <module '__main__' from '.../lung-tumor-segmentation-main/project/inference.py'>". I don't know how to solve the problem. Can you help me? Thanks a lot! #1
I added some fixes that should help - I added package versions (so you might have to re install some packages) and also added an important line to inference.py.
Try pulling the current code and recreating your python env to contain the correct package versions.
Hi there,
I added some fixes that should help - I added package versions (so you might have to re install some packages) and also added an important line to inference.py.
Try pulling the current code and recreating your python env to contain the correct package versions.
Let me know it solved the issue.