文件名称:移动语义细分:移动设备中的实时语义细分
文件大小:119KB
文件格式:ZIP
更新时间:2024-02-23 21:02:20
android python ios deep-learning tensorflow
移动设备中的实时语义分割 该项目是用于移动实时应用的语义分割的示例项目。 该体系结构受和启发。 )用作数据集。 该项目的目标是在移动设备中以合理的精度和速度检测头发片段。 目前,它达到0.89 IoU。 有关速度与准确性的更多信息,请参见。 应用范例 安卓(TODO) 要求 Python 3.8 pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html 适用于iOS应用的CoreML。 关于模型 目前,此存储库中只有一个模型 。 作为典型的U-Net架构,它具有编码器和解码器部分,这些部分由MobileNets提出的深度转换模块组成。 输入图像被编码为1/32大小,然后解码为1/2。 最后,它会对结果评分并使其达到原始大小。 训练步骤 资料准备 LFW提供数据。 要获取蒙版图像,请参阅 。 获得图像和蒙版后,如下所示放置面部和蒙版的图像。 data/ lfw/ raw/ images/ 0001.jpg
【文件预览】:
mobile-semantic-segmentation-master
----tmp()
--------.keep(0B)
----src()
--------.envrc(86B)
--------mylib()
--------run_convert_coreml.py(647B)
--------run_train.py(6KB)
--------run_eval.py(2KB)
--------params()
--------mobile_seg()
----Dockerfile(385B)
----requirements.txt(1KB)
----LICENSE(1KB)
----assets()
--------prediction.png(65KB)
----README.md(3KB)
----.gitignore(1KB)