【文件属性】:
文件名称:DeepLab-pt:DeepLabV3 plus使用PyTorch的实现
文件大小:12KB
文件格式:ZIP
更新时间:2021-04-21 08:16:23
training pytorch deeplab-v3-plus efficientnetv2 Python
DeepLabV3 +
使用PyTorch的语义分割实现
火车
运行python train.py进行培训
数据集结构(类似于CamVid数据集)
├── Dataset folder
├── train
├── 1111.png
├── 2222.png
├── train_labels
├── 1111_L.png
├── 2222_L.png
├── class_dict.csv
笔记
默认功能提取器是
更改培训配置,更改utils/config.py参数
默认的损失函数是weighted cross entropy
【文件预览】:
DeepLab-pt-master
----main.py(4KB)
----nets()
--------nn.py(12KB)
----LICENSE(11KB)
----README.md(696B)
----utils()
--------dataset.py(1KB)
--------util.py(6KB)
--------config.py(159B)
网友评论
- 毫无用处骗积分