跑通pytorch UNet++
2019-12-25 本文已影响0人
zelda2333
问题：
没有找到cv2
解决
应该安装opencv-python
pip install opencv-python
运行
python train.py --dataset dsb2018_96 --arch NestedUNet
问题
解决
from sklearn.externals import joblib
问题
解决，一张image，对应一张mask
问题：
解决：
未解决。。嘤嘤嘤太难了，还是换数据集吧。。
问题：
训练时GPU爆掉了。。
把bach_size调小，别瞎动别的。。人家写的挺好的
