'--testsize' parameter in test_SeaNet.py

[mirzaeihossein]

Dear Gongyang Li

Thanks for your fantastic paper.

I have a question about your code.
In "test_SeaNet.py" you define the ''testsize" parameter, and its default value is 288.
I change it to 576 or 144, but there are errors like this:

File "test_SeaNet.py", line 42, in
res, s34, s5, s12_sig, s34_sig, s5_sig, edge1, edge2 = model(image)
File "/home/hossein/miniconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/hossein/SeaNet-main/model/SeaNet_models.py", line 305, in forward
s12, s34, s5 = self.prediction_decoder(conv5, conv34, conv12)
File "/home/hossein/miniconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/hossein/SeaNet-main/model/SeaNet_models.py", line 256, in forward
x34_decoder = self.decoder34(torch.cat([x5_decoder, x34], 1))
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 48 but got size 46 for tensor number 1 in the list.

Is this code prepared just for (288,288,3)?
Is it possible to use this code for other dimensions? How?

Thanks a lot.

[MathLee]

Thank you for your attention to our work. May I ask which feature has a size of 46? It's x5_ Decoder or x34?