提交 4dc59f4c 编写于 作者: C chenguowei01

update README.md

上级 63ba4f49
...@@ -49,9 +49,10 @@ python video_infer.py --model_dir pretrained_weights/humanseg_lite_epxort ...@@ -49,9 +49,10 @@ python video_infer.py --model_dir pretrained_weights/humanseg_lite_epxort
# 对人像视频进行分割处理 # 对人像视频进行分割处理
python video_infer.py --model_dir pretrained_weights/humanseg_lite_epxort \ python video_infer.py --model_dir pretrained_weights/humanseg_lite_epxort \
----video_path data/video_test.mp4 --video_path data/video_test.mp4
``` ```
## 训练 ## 训练
使用下述命令基于与训练模型进行Fine-tuning,请确保选用的模型结构`model_type`与模型参数`pretrained_weights`匹配。 使用下述命令基于与训练模型进行Fine-tuning,请确保选用的模型结构`model_type`与模型参数`pretrained_weights`匹配。
```bash ```bash
...@@ -63,7 +64,7 @@ python train.py --model_type HumanSegMobile \ ...@@ -63,7 +64,7 @@ python train.py --model_type HumanSegMobile \
--pretrained_weights pretrained_weights/humanseg_Mobile \ --pretrained_weights pretrained_weights/humanseg_Mobile \
--batch_size 8 \ --batch_size 8 \
--learning_rate 0.001 \ --learning_rate 0.001 \
--num_epochs 10 \ --num_epochs 10
``` ```
其中参数含义如下: 其中参数含义如下:
* `--model_type`: 模型类型,可选项为:HumanSegServer、HumanSegMobile和HumanSegLite * `--model_type`: 模型类型,可选项为:HumanSegServer、HumanSegMobile和HumanSegLite
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册