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

test_urpc.py #28

Closed 18677064404 closed 3 years ago

18677064404 commented 3 years ago

Hi, Is there provide a 2D "test_urpc.py". I try to use your work:"train_uncertainty_rectified_pyramid_consistency_2D.py", but I don't know how to test it. Thanks!

Luoxd1996 commented 3 years ago

Hi, Please check this line. Best, Xiangde.

18677064404 commented 3 years ago

Thanks for your reply.

abcxubu commented 2 years ago

Thanks for your reply.

Hi, Did you successfully run the 2D 'test_urpc.py' ? I met the problem 'Traceback (most recent call last): File "/home/b417/xb/5_SSL4MIS-master/SSL4MIS-master/code/test_2D_fully.py", line 121, in metric = Inference(FLAGS) File "/home/b417/xb/5_SSL4MIS-master/SSL4MIS-master/code/test_2D_fully.py", line 110, in Inference case, net, test_save_path, FLAGS) File "/home/b417/xb/5_SSL4MIS-master/SSL4MIS-master/code/test_2D_fully.py", line 64, in test_single_volume out_main, dim=1), dim=1).squeeze(0) TypeError: softmax() received an invalid combination of arguments - got (tuple, dim=int), but expected one of: