上传者: 42136477
|
上传时间: 2022-03-07 19:11:36
|
文件大小: 4.42MB
|
文件类型: -
人为分割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