PIDNet icon indicating copy to clipboard operation
PIDNet copied to clipboard

为啥有的图片可以运行,有的图片会报错呢

Open tcoln opened this issue 10 months ago • 2 comments

python tools/custom.py --a 'pidnet-s' --p './pretrained_models/cityscapes/PIDNet_S_Cityscapes_test.pt' --t '.png'

报错信息: Traceback (most recent call last): File "tools/custom.py", line 94, in pred = model(img) File "/mnt/data/glt/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/mnt/data/glt/safep/ponding/PIDNet/tools/../models/pidnet.py", line 149, in forward x_d = x_d + F.interpolate( RuntimeError: The size of tensor a (209) must match the size of tensor b (208) at non-singleton dimension 2

报错图片:链接:https://pan.baidu.com/s/1tloW25TLcR69xglzwcIPtQ 提取码:1234

tcoln avatar Mar 25 '24 11:03 tcoln

您好,请问您在cityscapes数据集上训练的时候有没有遇到这种情况以及如何解决的:loss不收敛且semantic loss 和 sb loss一直为0 ?如果没有遇到的话,请问您训练时的参数如何设置的? 微信图片_20240331103215

qmm1234 avatar Mar 31 '24 09:03 qmm1234

same error

sym330 avatar Jul 02 '24 08:07 sym330