【文件属性】：
文件名称：PyTorch-ENet:ENet的PyTorch实施
文件大小：7.69MB
文件格式：ZIP
更新时间：2021-05-05 13:17:00
pytorch enet cityscape camvid Python
PyTorch-ENet
ENet的PyTorch（v1.1.0）实现，从作者创建的lua-torch实现移植而来。
此实现已在CamVid和Cityscapes数据集上进行了测试。 当前，可获得在CamVid和Cityscapes中训练的模型的预训练版本。
数据集
1班
输入分辨率
批量大小
时代
平均IoU（％）
GPU内存（GiB）
训练时间（小时） 2
11
480x360
10
300
51.08 3
4.2
1个
19
1024x512
4
300
59.03 4
5.4
20
1当涉及到类数时，总是将无效/未标记的类排除在外。 2这些仅供参考。 实施，数据集和硬件更改可能导致截然不同的结果。 参考硬件：Nvidia GTX 1070和AMD Ryzen 5 3600 3.6GHz。 您还可以训练约100个纪元，并获得相似的平均IoU（±2
【文件预览】：
PyTorch-ENet-master
----train.py(2KB)
----save()
--------ENet_CamVid()
--------README.md(2KB)
--------ENet_Cityscapes()
----models()
--------enet.py(22KB)
----test.py(2KB)
----utils.py(4KB)
----main.py(11KB)
----requirements.txt(176B)
----LICENSE(1KB)
----README.md(5KB)
----metric()
--------confusionmatrix.py(4KB)
--------metric.py(245B)
--------__init__.py(141B)
--------iou.py(4KB)
----args.py(4KB)
----transforms.py(3KB)
----data()
--------camvid.py(5KB)
--------utils.py(6KB)
--------cityscapes.py(6KB)
--------__init__.py(98B)
--------README.md(1KB)
----.gitignore(1KB)