search

HiLab-git / SSL4MIS

Semi Supervised Learning for Medical Image Segmentation, a collection of literature reviews and code implementations.
MIT License
2.19k stars 387 forks source link

No module named 'val_unet_2D_dv' #3

Closed hannhu closed 3 years ago

hannhu commented 3 years ago

Hi, I found some problem in 2ed step as following: Traceback (most recent call last): File "train_unet_2D_dv_fully_supervised.py", line 23, in <module> from val_unet_2D_dv import test_single_volume ModuleNotFoundError: No module named 'val_unet_2D_dv' I cant find the module "val_unet_2D_dv" in your repository or pypi, How can I get it?

hannhu commented 3 years ago

I found the function of the same name in code/val_unet_2D.py. Is it right?

Luoxd1996 commented 3 years ago

Hi hannhu, Sorry, it's my fault, I have updated it just now. Please clone or pull the repo again and re-try it. Best wishes; Xiangde Luo.

hannhu commented 3 years ago

Hi hannhu, Sorry, it's my fault, I have updated it just now. Please clone or pull the repo again and re-try it. Best wishes; Xiangde Luo.

thx, but get new error Traceback (most recent call last): File "train_unet_2D_dv_fully_supervised.py", line 225, in <module> train(args, snapshot_path) File "train_unet_2D_dv_fully_supervised.py", line 74, in train model = net_factory(net_type=args.model, in_chns=1, class_num=num_classes) File "/home/huhan/python/semi-seg/SSL4MIS/code/networks/net_factory.py", line 9, in net_factory net = UNet_DS(in_chns=in_chns, class_num=class_num).cuda() File "/home/huhan/python/semi-seg/SSL4MIS/code/networks/unet.py", line 224, in __init__ self.decoder = Decoder_DS(params) File "/home/huhan/python/semi-seg/SSL4MIS/code/networks/unet.py", line 164, in __init__ self.feature_noise = FeatureNoise() NameError: name 'FeatureNoise' is not defined I cant find it's definition anywhere.

Luoxd1996 commented 3 years ago

Hi, hannhu, The UNet_DS is a new baseline for this project, maybe there are some bugs. I am testing these code, but limited by the GPUs, I need some time to finish it. You can use the UNet as baseline for your task or tell me these bugs, I will fix them as soon as possible. After I finished these code testing, I will email you. By the way, the error of "FeatureNoise" I have fixed. Best wishes, Xiangde Luo.