brain-segmentation-pytorch
brain-segmentation-pytorch copied to clipboard
U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI
Hello, Running python script cause to an error: ``` reading validation images... Traceback (most recent call last): File "brain-segmentation-pytorch/inference.py", line 184, in main(args) File "brain-segmentation-pytorch/inference.py", line 22, in main loader...
When changing the kwarg (`in_channels`) from 3 to 2, like ``` torch.hub.load('mateuszbuda/brain-segmentation-pytorch', 'unet', in_channels=2, out_channels=1, init_features=32, pretrained=True) ``` this error occurs: ``` ...module.py", line 1044, in load_state_dict raise RuntimeError('Error(s) in...
Thanks for ur sharing and ur work is really cool! I have a puzzle about ur code in dataset.py , what is ur point in linë 48 and linë 49?...
Hi, I am trying to train the network on a small dataset of brain MRI images. I have sliced out the images and masks following the format of your dataset....
Hi, I retrained your model on my own dataset; However, I see the Dice Loss is 1 because the prediction is totally wrong and the segmented area is really small....
Repository is no longer compatible with scipy as of version 1.3 ; this is due to a deprecated call to scipy.misc.toimage See: https://docs.scipy.org/doc/scipy/reference/release.1.3.0.html?highlight=toimage There are two call in logger.py to...
Hi, I am testing your code and ended up with the following error. My test data has 11 patients with around 800 .tifs (400 images, 400 masks) per patient. I...
manifest for nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 not found: manifest unknown: manifest unknown
while trying to build dockerfile getting this message: manifest for nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 not found: manifest unknown: manifest unknown