HiLab-git / SSL4MIS

Semi Supervised Learning for Medical Image Segmentation, a collection of literature reviews and code implementations.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

test_urpc.py

18677064404 opened this issue · comments

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!

Hi,
Please check this line.
Best,
Xiangde.

Thanks for your reply.

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:

  • (Tensor input, name dim, *, torch.dtype dtype)
  • (Tensor input, int dim, torch.dtype dtype)'. I am wondering if you know how to solve it
    1
    2
    .