code for prediction and visualization on segmentation output

[amiltonwong]

Hi, @dmsm,

Could you also provide the code for prediction and visualization on segmentation output?

Thanks~

[Dengzhi-USTC]

Hi, @dmsm @amiltonwong , Can you retrain this code? I have been waiting in the line 78 of the hodgeautograd.py( ctx_eigenvectors.copy_(torch.from_numpy(reshaped_eigenvectors).to(device))) when run on the cpu; Did you meet this issure?

Thanks

[Dengzhi-USTC]

Hi, @dmsm @amiltonwong , Can you retrain this code? I have been waiting in the line 78 of the hodgeautograd.py( ctx_eigenvectors.copy_(torch.from_numpy(reshaped_eigenvectors).to(device))) when run on the cpu; Did you meet this issure?

Thanks

I solved this problem, need considerate the version of the torch.multiprocess. (Add the torch.multiprocessing.set_start_method('spawn', force=True) is ok)

[ManjitKuma]

Have anyone tried to test and visualize the output of the model? I tried but not getting expected result .if anyone got then please give me some suggestion ..