文件名称：PyTorch-ENet:ENet的PyTorch实施
文件大小：7.69MB
文件格式：ZIP
更新时间：2024-05-25 07:03:40
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)