
基于Unet的城市景观与Kitti数据集应用研究
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究运用深度学习框架Unet,结合城市景观及Kitti数据集,探索图像分割技术在复杂道路环境下的应用优化。
使用PyTorch进行语义分割的UNet模型可以利用Cityscapes和Kitti数据集训练。在预言训练好模型并将其保存为MODEL.pth之后,可以通过命令行界面轻松测试图像上的输出蒙版。
要预测单个图像并将结果保存,请执行以下操作:
```
python predict.py -i image.jpg -o output.jpg
```
若需预测多幅图像并在不进行存储的情况下显示它们,则可以使用如下指令:
```
python predict.py -i image1.jpg image2.jpg --viz --no-save
```
全部评论 (0)
还没有任何评论哟~


