This repository includes the official project of Mask Guided (MG) Matting, presented in our paper: Mask Guided Matting via Progressive Refinement Network
Other
332
stars
48
forks
source link
RuntimeError: "upsample_nearest2d_out_frame" not implemented for 'Byte' #4
Thank you so much for this awesome work. I try training the model with a very small dataset and ran the infer.py. However, I got this error straight away RuntimeError: "upsample_nearest2d_out_frame" not implemented for 'Byte'
It seemed to be from
Hi @yucornetto,
Thank you so much for this awesome work. I try training the model with a very small dataset and ran the infer.py. However, I got this error straight away
RuntimeError: "upsample_nearest2d_out_frame" not implemented for 'Byte'
It seemed to be fromutil.py in get_unknown_tensor_from_pred(pred, rand_width, train_mode) 244 weight = torch.from_numpy(weight).cuda() 245 --> 246 weight = F.interpolate(weight, size=(H,W), mode='nearest')
Can you please give me a small guide on how to fix it? Thank you so much!