Skip to content

Commit 105efa4

Browse files
authored
Fix rotated detection data pipeline (#3627)
fix data pipeline
1 parent b3b13ba commit 105efa4

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

src/otx/algorithms/detection/configs/rotated_detection/efficientnetb2b_maskrcnn/data_pipeline.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
resize_cfg=dict(
2525
type="Resize",
2626
img_scale=__img_size,
27-
keep_ratio=True,
27+
keep_ratio=False,
2828
),
2929
enable_memcache=True, # Cache after resizing image & annotations
3030
),
@@ -54,7 +54,7 @@
5454
val_pipeline = [
5555
dict(
5656
type="LoadResizeDataFromOTXDataset",
57-
resize_cfg=dict(type="Resize", img_scale=__img_size, keep_ratio=True),
57+
resize_cfg=dict(type="Resize", img_scale=__img_size, keep_ratio=False),
5858
enable_memcache=True, # Cache after resizing image
5959
),
6060
dict(

src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/data_pipeline.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
resize_cfg=dict(
2323
type="Resize",
2424
img_scale=__img_size,
25-
keep_ratio=True,
25+
keep_ratio=False,
2626
),
2727
enable_memcache=True, # Cache after resizing image & annotations
2828
),
@@ -52,7 +52,7 @@
5252
val_pipeline = [
5353
dict(
5454
type="LoadResizeDataFromOTXDataset",
55-
resize_cfg=dict(type="Resize", img_scale=__img_size, keep_ratio=True),
55+
resize_cfg=dict(type="Resize", img_scale=__img_size, keep_ratio=False),
5656
enable_memcache=True, # Cache after resizing image
5757
),
5858
dict(
@@ -76,7 +76,7 @@
7676
img_scale=__img_size,
7777
flip=False,
7878
transforms=[
79-
dict(type="Resize", keep_ratio=True),
79+
dict(type="Resize", keep_ratio=False),
8080
dict(type="RandomFlip"),
8181
dict(type="Normalize", **__img_norm_cfg),
8282
dict(type="Pad", size_divisor=32),

src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/deployment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,5 +8,5 @@
88

99
backend_config = dict(
1010
# dynamic batch causes forever running openvino process
11-
model_inputs=[dict(opt_shapes=dict(input=[1, 3, 800, 1344]))],
11+
model_inputs=[dict(opt_shapes=dict(input=[1, 3, 1024, 1024]))],
1212
)

0 commit comments

Comments
 (0)