Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
ba365ae8
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
285
Star
8
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSeg
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
ba365ae8
编写于
5月 12, 2020
作者:
C
chenguowei01
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update README.md
上级
7892521b
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
12 addition
and
6 deletion
+12
-6
contrib/HumanSeg/README.md
contrib/HumanSeg/README.md
+12
-6
未找到文件。
contrib/HumanSeg/README.md
浏览文件 @
ba365ae8
...
...
@@ -18,14 +18,15 @@ $ pip install -r requirements.txt
## 模型
| 模型类型 | 预训练模型 | 导出模型 | 量化模型 | 说明 |
| --- | --- | --- | --- | --- |
| HumanSegServer |
[
humanseg_server
](
)
|
[
humanseg_server_export
](
)
|
[
humanseg_server_quant
](
)
| 服务端GPU环境 |
| HumanSegMobile |
[
humanseg_mobile
](
)
|
[
humanseg_mobile_export
](
)
|
[
humanseg_mobile_quant
](
)
| 小模型, 适合轻量级计算环境 |
| HumanSegLite |
[
humanseg_lite
](
)
|
[
humanseg_lite_export
](
)
|
[
humanseg_lite_quant
](
)
| 小模型, 适合轻量级计算环境 |
| HumanSegServer |
[
humanseg_server
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_server.zip
)
|
[
humanseg_server_export
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_server_export.zip
)
|
[
humanseg_server_quant
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_server_quant.zip
)
| 服务端GPU环境 |
| HumanSegMobile |
[
humanseg_mobile
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_mobile.zip
)
|
[
humanseg_mobile_export
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_mobile_export.zip
)
|
[
humanseg_mobile_quant
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_mobile_quant.zip
)
| 小模型, 适合轻量级计算环境 |
| HumanSegLite |
[
humanseg_lite
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_lite.zip
)
|
[
humanseg_lite_export
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_lite_export.zip
)
|
[
humanseg_lite_quant
](
https://paddleseg.bj.bcebos.com/humanseg/models/humanseg_lite_quant.zip
)
| 小模型, 适合轻量级计算环境 |
##
视频流分割
##
指定运行设备
```
bash
python video_infer.py
--model_dir
path/to/model_dir
export
CUDA_VISIBLE_DEVICES
=
0
```
当CUDA_VISIBLE_DEVICES变量有效时,使用相应的显卡进行计算,无效时使用CPU进行计算
## 准备训练数据
我们提供了一份demo数据集,通过运行以下代码进行下载,该数据集是从supervise.ly抽取的一个小数据集。
...
...
@@ -40,10 +41,15 @@ python data/download_data.py
python pretrained_weights/download_pretrained_weights.py
```
## 视频流分割
```
bash
python video_infer.py
--model_dir
path/to/model_dir
```
## 训练
使用下述命令进行训练
```
bash
CUDA_VISIBLE_DEVICES
=
0
&&
python train.py
--model_type
HumanSegMobile
\
python train.py
--model_type
HumanSegMobile
\
--save_dir
output/
\
--data_dir
data/mini_supervisely
\
--train_list
data/mini_supervisely/train.txt
\
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录