【文件属性】：
文件名称：Unet：具有城市景观和kitti数据集的Unet
文件大小：12.66MB
文件格式：ZIP
更新时间：2021-03-02 02:15:51
Python
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.jpg --viz --no-save
> python predict.py -h
usage: predict.py [-h] [--model FILE] --input INPUT [INPUT ...]
[--output INPUT [INPUT