From 0e58790c962171549e20882ef4c1e12391588408 Mon Sep 17 00:00:00 2001 From: chenguowei01 Date: Fri, 17 Jul 2020 11:50:03 +0800 Subject: [PATCH] update README.md --- dygraph/README.md | 31 ++++++++----------------------- 1 file changed, 8 insertions(+), 23 deletions(-) diff --git a/dygraph/README.md b/dygraph/README.md index 20728e6b..fd9af0d0 100644 --- a/dygraph/README.md +++ b/dygraph/README.md @@ -1,43 +1,28 @@ # 动态图执行 -## 数据集设置 -``` -data_dir='data/path' -train_list='train/list/path' -val_list='val/list/path' -test_list='test/list/path' -num_classes=number/of/dataset/classes -``` - ## 训练 ``` python3 train.py --model_name UNet \ ---data_dir $data_dir \ ---train_list $train_list \ ---val_list $val_list \ ---num_classes $num_classes \ +--dataset OpticDiscSeg \ --input_size 192 192 \ ---num_epochs 4 \ +--num_epochs 10 \ --save_interval_epochs 1 \ +--do_eval \ --save_dir output ``` ## 评估 ``` python3 val.py --model_name UNet \ ---data_dir $data_dir \ ---val_list $val_list \ ---num_classes $num_classes \ +--dataset OpticDiscSeg \ --input_size 192 192 \ ---model_dir output/epoch_1 +--model_dir output/best_model ``` ## 预测 ``` python3 infer.py --model_name UNet \ ---data_dir $data_dir \ ---test_list $test_list \ ---num_classes $num_classes \ ---input_size 192 192 \ ---model_dir output/epoch_1 +--dataset OpticDiscSeg \ +--model_dir output/best_model \ +--input_size 192 192 ``` -- GitLab