icey-zhang / SuperYOLO

SuperYOLO is accepted by TGRS
323 stars 54 forks source link

L1loss里的输入参数问题 #103

Closed leileilei2000 closed 4 months ago

leileilei2000 commented 6 months ago

sr_loss = 0.1*(torch.nn.L1Loss()(output_sr[:,0:3,:,:,],image)+torch.nn.L1Loss()(output_sr[:,3:,:,:,],ir_image[:,0:1,:,:,])) 请问output_sr、image、ir_image是在哪里引入的?我没有找到具体的位置 我在上面找到了pred,outputsr, = model(imgs,irs,opt.input_mode),但是model有好多个我不知道具体是哪个,所以找不到output_sr和imgs,irs的值是什么