MigrationNet:使用PyTorch进行地下管线解释
基于PyTorch中的构建。
用法
注意:使用Python 3
预言
您可以使用我们的数据集轻松测试输出:
python predict_mat.py -i path/to/test -o predict.png -m path/to/checkpoint
训练
python train_mat.py -f path/to/checkpoint -e 200 -b 1 -l 0.000005 -s 0.25 -x path/to/train -y path/to/gt
2022-05-11 19:12:40
46.86MB
Python
1