【文件属性】:
文件名称:Human-Segmentation
文件大小:2.43MB
文件格式:ZIP
更新时间:2021-04-11 04:56:13
Python
人为分割PyTorch
在PyTorch中实现的人体分割,/代码和。
支持的网络
:骨干 (所有aphas和扩张), (所有num_layers)
:骨干网 (num_layers = 18,34,50,101),
:骨干网 (num_layers = 18)
:主干网 (num_layers = 18、34、50、101)
ICNet :主干网ResNetV1 (num_layers = 18、34、50、101)
要评估体系结构,内存,转发时间(以cpu或gpu表示),参数数量以及网络的FLOP数量,请使用以下命令:
python measure_model.py
数据集
人像分割(人/背景)
自动人像分割以实现图像风格化:1800张图像
监督人:5711张图片
放
在此存储库中使用了Python3.6.x。
克隆存储库:
git clone --re
【文件预览】:
Human-Segmentation-main
----train.py(3KB)
----base()
--------base_trainer.py(8KB)
--------__init__.py(343B)
--------base_model.py(5KB)
--------base_data_loader.py(2KB)
--------base_inference.py(5KB)
----requirements.txt(604B)
----dataset()
--------train_mask.txt(804KB)
--------valid_mask.txt(89KB)
--------create_pairs.py(4KB)
----inference_webcam.py(1KB)
----README.md(5KB)
----measure_model.py(3KB)
----inference_video.py(5KB)
----backgrounds()
--------2.jpg(19KB)
--------6.jpg(436KB)
--------1.png(204KB)
--------4.jpg(923KB)
--------5.jpg(6KB)
--------7.jpg(823KB)
--------3.jpg(13KB)