awccnet / awcc-net Goto Github PK

Hello author, thank you for your excellent work. May I ask if you could share the loss function and training code? Thank you！

After configuring the environment, when I run the following command: python main.py --dir_path /root/Code/AWCC/AWCC-Net/test/ --checkpoint /root/Code/AWCC/AWCC-Net/weight/best.pth, the program runs for a few seconds and then throws the following error:

0it [00:00, ?it/s]
/root/miniconda3/envs/AWCC1/lib/python3.10/site-packages/numpy/core/fromnumeric.py:3504: RuntimeWarning: Mean of empty slice.
return _methods._mean(a, axis=axis, dtype=dtype,
/root/miniconda3/envs/AWCC1/lib/python3.10/site-packages/numpy/core/_methods.py:129: RuntimeWarning: invalid value encountered in scalar divide
ret = ret.dtype.type(ret / rcount)
mse: nan, mae: nan

How can I resolve this issue?

Where can i get the processed JHU dataset the project used?