Update README.md

yahuiwei123 · Dec 20, 2023 · 5b26b28 · 5b26b28
1 parent 36b7283
commit 5b26b28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -15,7 +15,7 @@ python train.py --batch_size 8 --step 8 --learning_rate 0.01 --num_epochs 100 -o
 
 ##### 模型预测
 ```python
-python predict.py --image_path './test/img' --step 8 --output_size (128, 128) --output_dir './test/out'
+python predict.py --image_path './test/img' --step 8 --output_size (480, 480) --output_dir './test/out'
 ```
 #### 模型细节
 + 对模型中的如下模块进行了替换