search

Linfeng-Tang / SeAFusion

The code of " Image fusion in the loop of high-level vision tasks: A semantic-aware real-time infrared and visible image fusion network"
MIT License
184 stars 34 forks source link

evaluate.py报错 #39

Open liushh39 opened 7 months ago

liushh39 commented 7 months ago

line 188, in evaluate probs_torch += prob RuntimeError: The size of tensor a (640) must match the size of tensor b (520) at non-singleton dimension 3

Linfeng-Tang commented 7 months ago

是不是用了其他数据集导致的呀