diff --git a/dygraph/README.md b/dygraph/README.md index 20728e6bb5321e78122b9e925544419e9ec4483c..fd9af0d0dac1c2011479de4cab6929e34c18a515 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 ```