diff --git a/contrib/HumanSeg/README.md b/contrib/HumanSeg/README.md
index 52a57590dd6bce71c785cc10d37b74357d0f15ce..d1da18c0b7d70dc866ec299917b2096c38e116c9 100644
--- a/contrib/HumanSeg/README.md
+++ b/contrib/HumanSeg/README.md
@@ -44,10 +44,10 @@ python data/download_data.py
 ## 快速体验视频流人像分割
 ```bash
 # 通过电脑摄像头进行实时分割处理
-python video_infer.py --model_dir pretrained_weights/humanseg_lite_export
+python video_infer.py --model_dir pretrained_weights/humanseg_lite_inference
 
 # 对人像视频进行分割处理
-python video_infer.py --model_dir pretrained_weights/humanseg_lite_export --video_path data/video_test.mp4
+python video_infer.py --model_dir pretrained_weights/humanseg_lite_inference --video_path data/video_test.mp4
 ```
 
 **NOTE**:
@@ -62,7 +62,7 @@ python train.py --model_type HumanSegMobile \
 --data_dir data/mini_supervisely \
 --train_list data/mini_supervisely/train.txt \
 --val_list data/mini_supervisely/val.txt \
---pretrained_weights pretrained_weights/humanseg_mobile \
+--pretrained_weights pretrained_weights/humanseg_mobile_ckpt \
 --batch_size 8 \
 --learning_rate 0.001 \
 --num_epochs 10 \
diff --git a/contrib/HumanSeg/nets/__init__.py b/contrib/HumanSeg/nets/__init__.py
index d44308d7718859977bdb7cbc70d4a0d0d0b2b97f..cab1682745084e736992e4aa5e555db6bc1d5c53 100644
--- a/contrib/HumanSeg/nets/__init__.py
+++ b/contrib/HumanSeg/nets/__init__.py
@@ -1,6 +1,5 @@
 from .backbone import mobilenet_v2
 from .backbone import xception
-from .unet import UNet
 from .deeplabv3p import DeepLabv3p
 from .shufflenet_slim import ShuffleSeg
 from .hrnet import HRNet