| 123456789101112131415161718192021222324252627282930313233343536373839404142 |
- Global:
- model: MaskFormer_small
- mode: check_dataset # check_dataset/train/evaluate/predict
- dataset_dir: "/paddle/dataset/paddlex/seg/seg_optic_examples"
- device: gpu:0,1,2,3
- output: "output"
- CheckDataset:
- convert:
- enable: False
- src_dataset_type: null
- split:
- enable: False
- train_percent: null
- val_percent: null
- Train:
- epochs_iters: 500
- num_classes: 2
- batch_size: 2
- learning_rate: 0.01
- pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/MaskFormer_small_pretrained.pdparams
- warmup_steps: 0
- resume_path: null
- log_interval: 10
- eval_interval: 100
- input_shape: [1, 3, 512, 512] # export the model with fixed input shape
- Evaluate:
- weight_path: "output/best_model/model.pdparams"
- log_interval: 10
- Export:
- weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/MaskFormer_small_pretrained.pdparams
- input_shape: [1, 3, 512, 512]
- Predict:
- batch_size: 1
- model_dir: "output/best_model/inference"
- input: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_semantic_segmentation_001.jpg"
- kernel_option:
- run_mode: paddle
|
