Skip to content

Commit 52a13c4

Browse files
restore test case for detr-resnet50 (#3204)
* restore test case for detr-resnet50 * rm detr from excluded models
1 parent ffb9fb2 commit 52a13c4

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

demos/tests/cases.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1169,13 +1169,13 @@ def single_option_cases(key, *args):
11691169
TestCase(options={'-at': 'yolov4', '-m': ModelArg('yolo-v4-tiny-tf')}),
11701170
TestCase(options={'-at': 'yolof', '-m': ModelArg('yolof')}),
11711171
*combine_cases(
1172-
TestCase(options={'--architecture_type': 'detr'}), # TODO detr-resnet50 model fails to convert on 2022.1 package
1172+
TestCase(options={'--architecture_type': 'detr'}),
11731173
[
1174-
#TestCase(options={'-m': ModelArg('detr-resnet50')}),
1175-
#TestCase(options={'-m': ModelFileArg('detr-resnet50', 'detr-resnet50.onnx'),
1176-
# '--reverse_input_channels': None,
1177-
# '--mean_values': ['123.675', '116.28', '103.53'],
1178-
# '--scale_values': ['58.395', '57.12', '57.375']}),
1174+
TestCase(options={'-m': ModelArg('detr-resnet50')}),
1175+
TestCase(options={'-m': ModelFileArg('detr-resnet50', 'detr-resnet50.onnx'),
1176+
'--reverse_input_channels': None,
1177+
'--mean_values': ['123.675', '116.28', '103.53'],
1178+
'--scale_values': ['58.395', '57.12', '57.375']}),
11791179
]
11801180
),
11811181
*combine_cases(

tools/model_tools/src/openvino/model_zoo/_configuration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
from openvino.model_zoo.download_engine import cache, file_source, postprocessing, validation
2525

2626
RE_MODEL_NAME = re.compile(r'[0-9a-zA-Z._-]+')
27-
EXCLUDED_MODELS = ['detr-resnet50']
27+
EXCLUDED_MODELS = []
2828

2929
class ModelFile:
3030
def __init__(self, name, size, checksum, source):

0 commit comments

Comments
 (0)