UNet：使用PyTorch进行语义分割 使用Cityscapes和Kitti数据集 用法 注意：使用Python 3.6或更高版本 预言 训练好模型并将其保存到MODEL.pth后，您可以通过CLI轻松测试图像上的输出蒙版。 要预测单个图像并保存，请执行以下操作： python predict.py -i image.jpg -o output.jpg 要预测多幅图像并显示它们而不保存它们，请执行以下操作： python predict.py -i image1.jpg image2