代码来自 U-Net: Convolutional Networks for Biomedical Image Segmentation
已经运行训练生成.pth权重文件了,希望能够采用命令行预测指定路径下的所有图片并将结果保存到目标路径,但是怎么改路径都报错。
希望python predict.py -i data_predict/ -m checkpoints/checkpoint_epoch5.pth -o result/ 这个命令可以正确执行应该如何修改?
源码地址:https://github.com/milesial/Pytorch-UNet
相似问题