You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
1.9 KiB
74 lines
1.9 KiB
# Basic configurations of SARShip dataset |
|
|
|
datasets: |
|
train: !Node |
|
type: VOCDetDataset |
|
args: |
|
data_dir: ./test_tipc/data/sarship/ |
|
file_list: ./test_tipc/data/sarship/train.txt |
|
label_list: ./test_tipc/data/sarship/labels.txt |
|
shuffle: True |
|
eval: !Node |
|
type: VOCDetDataset |
|
args: |
|
data_dir: ./test_tipc/data/sarship/ |
|
file_list: ./test_tipc/data/sarship/eval.txt |
|
label_list: ./test_tipc/data/sarship/labels.txt |
|
shuffle: False |
|
transforms: |
|
train: |
|
- !Node |
|
type: DecodeImg |
|
- !Node |
|
type: RandomDistort |
|
- !Node |
|
type: RandomExpand |
|
- !Node |
|
type: RandomCrop |
|
- !Node |
|
type: RandomHorizontalFlip |
|
- !Node |
|
type: BatchRandomResize |
|
args: |
|
target_sizes: [320, 352, 384, 416, 448, 480, 512, 544, 576, 608] |
|
interp: RANDOM |
|
- !Node |
|
type: Normalize |
|
args: |
|
mean: [0.485, 0.456, 0.406] |
|
std: [0.229, 0.224, 0.225] |
|
- !Node |
|
type: ArrangeDetector |
|
args: ['train'] |
|
eval: |
|
- !Node |
|
type: DecodeImg |
|
- !Node |
|
type: Resize |
|
args: |
|
target_size: 608 |
|
interp: CUBIC |
|
- !Node |
|
type: Normalize |
|
args: |
|
mean: [0.485, 0.456, 0.406] |
|
std: [0.229, 0.224, 0.225] |
|
- !Node |
|
type: ArrangeDetector |
|
args: ['eval'] |
|
download_on: False |
|
download_url: https://paddlers.bj.bcebos.com/datasets/sarship.zip |
|
download_path: ./test_tipc/data/ |
|
|
|
num_epochs: 10 |
|
train_batch_size: 4 |
|
save_interval_epochs: 5 |
|
log_interval_steps: 4 |
|
save_dir: ./test_tipc/output/det/ |
|
learning_rate: 0.0005 |
|
use_vdl: False |
|
resume_checkpoint: '' |
|
train: |
|
pretrain_weights: COCO |
|
warmup_steps: 0 |
|
warmup_start_lr: 0.0 |