-
Notifications
You must be signed in to change notification settings - Fork 447
/
Copy pathrtmdet_inst_tiny_tv.yaml
136 lines (131 loc) · 3.92 KB
/
rtmdet_inst_tiny_tv.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
model:
class_path: otx.algo.instance_segmentation.rtmdet_inst.MMDetRTMDetInstTiny
init_args:
label_info: 80
optimizer:
class_path: torch.optim.SGD
init_args:
lr: 0.001
momentum: 0.9
weight_decay: 0.0001
scheduler:
class_path: otx.core.schedulers.LinearWarmupSchedulerCallable
init_args:
num_warmup_steps: 20
main_scheduler_callable:
class_path: lightning.pytorch.cli.ReduceLROnPlateau
init_args:
mode: max
factor: 0.1
patience: 9
monitor: val/map_50
min_lr: 4e-06
engine:
task: INSTANCE_SEGMENTATION
device: auto
callback_monitor: val/map_50
data: ../_base_/data/torchvision_base.yaml
overrides:
precision: 16
max_epochs: 100
gradient_clip_val: 35.0
data:
task: INSTANCE_SEGMENTATION
config:
stack_images: true
data_format: coco_instances
include_polygons: true
train_subset:
batch_size: 4
num_workers: 8
transforms:
- class_path: otx.core.data.transform_libs.torchvision.CachedMosaic
init_args:
img_scale:
- 640
- 640
pad_val: 114.0
max_cached_images: 20
random_pop: false
- class_path: otx.core.data.transform_libs.torchvision.RandomResize
init_args:
scale:
- 1280
- 1280
ratio_range:
- 0.5
- 2.0
keep_ratio: true
transform_bbox: true
transform_mask: true
- class_path: otx.core.data.transform_libs.torchvision.RandomCrop
init_args:
crop_size:
- 640
- 640
- class_path: otx.core.data.transform_libs.torchvision.YOLOXHSVRandomAug
- class_path: otx.core.data.transform_libs.torchvision.RandomFlip
init_args:
prob: 0.5
- class_path: otx.core.data.transform_libs.torchvision.Pad
init_args:
pad_to_square: true
pad_val: 114
transform_mask: true
- class_path: otx.core.data.transform_libs.torchvision.CachedMixUp
init_args:
img_scale:
- 640
- 640
ratio_range:
- 1.0
- 1.0
max_cached_images: 10
random_pop: false
pad_val:
- 114
- 114
- 114
prob: 0.5
- class_path: otx.core.data.transform_libs.torchvision.FilterAnnotations
init_args:
min_gt_bbox_wh:
- 1
- 1
is_numpy_to_tvtensor: true
val_subset:
batch_size: 1
num_workers: 4
transforms:
- class_path: otx.core.data.transform_libs.torchvision.Resize
init_args:
scale:
- 640
- 640
keep_ratio: true
transform_bbox: false
transform_mask: false
- class_path: otx.core.data.transform_libs.torchvision.Pad
init_args:
pad_to_square: true
pad_val: 114
transform_mask: false
is_numpy_to_tvtensor: true
test_subset:
batch_size: 1
num_workers: 4
transforms:
- class_path: otx.core.data.transform_libs.torchvision.Resize
init_args:
scale:
- 640
- 640
keep_ratio: true
transform_bbox: false
transform_mask: false
- class_path: otx.core.data.transform_libs.torchvision.Pad
init_args:
pad_to_square: true
pad_val: 114
transform_mask: false
is_numpy_to_tvtensor: true