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

update README.md

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