dddb11 / MVSS-Net

Unofficial implementation of MVSS-Net (ICCV 2021) with Pytorch including training code.
59 stars 10 forks source link

fixed cv2.resize error and type error of labels, the modified files are in datasets_v2.py and train_base_v2.py. #12

Closed IrumaMegumi closed 9 months ago

IrumaMegumi commented 9 months ago

In your original /datasets/dataset.py file, I added a line. Without it will cause a cv2.resize error.

image
IrumaMegumi commented 9 months ago

Position is near line 493.

SunnyHaze commented 9 months ago

Thanks for you clarify of the bugging line and fixing this bug, and sorry for our negligence of this bug. Besides, I deleted the extra train_base_v2.py file manually to keep the main branch clean. If you want to keep it in your own repository, you can still find it in previous commits and copy it offline.

IrumaMegumi commented 9 months ago

OK,thanks. If you like, you can cantact me at QQ:2286453531