diff --git a/demos/3d_segmentation_demo/python/models.lst b/demos/3d_segmentation_demo/python/models.lst index f336fb4090f..7a3c329c23e 100644 --- a/demos/3d_segmentation_demo/python/models.lst +++ b/demos/3d_segmentation_demo/python/models.lst @@ -1,2 +1,2 @@ # This file can be used with the --list option of the model downloader. -brain-tumor-segmentation-???? +brain-tumor-segmentation-0002 diff --git a/demos/object_detection_demo/python/README.md b/demos/object_detection_demo/python/README.md index aac3f18dab4..a789279766e 100644 --- a/demos/object_detection_demo/python/README.md +++ b/demos/object_detection_demo/python/README.md @@ -129,7 +129,6 @@ omz_converter --list models.lst - ultra-lightweight-face-detection-rfb-320 - ultra-lightweight-face-detection-slim-320 * architecture_type = yolo - - mobilefacedet-v1-mxnet - mobilenet-yolo-v4-syg - person-vehicle-bike-detection-crossroad-yolov3-1020 - yolo-v1-tiny-tf diff --git a/demos/object_detection_demo/python/models.lst b/demos/object_detection_demo/python/models.lst index 433a5a83a4a..bda77bb2dd1 100644 --- a/demos/object_detection_demo/python/models.lst +++ b/demos/object_detection_demo/python/models.lst @@ -44,7 +44,6 @@ vehicle-license-plate-detection-barrier-???? ultra-lightweight-face-detection-rfb-320 ultra-lightweight-face-detection-slim-320 # For --architecture_type=yolo -mobilefacedet-v1-mxnet mobilenet-yolo-v4-syg person-vehicle-bike-detection-crossroad-yolov3-1020 yolo-v1-tiny-tf diff --git a/demos/tests/cases.py b/demos/tests/cases.py index aa71e9d0b0a..62b5d77826b 100644 --- a/demos/tests/cases.py +++ b/demos/tests/cases.py @@ -729,8 +729,9 @@ def single_option_cases(key, *args): )), PythonDemo(name='3d_segmentation_demo', device_keys=['-d'], test_cases=combine_cases( - TestCase(options={'-m': ModelArg('brain-tumor-segmentation-0001'), - '-o': '.'}), + TestCase(options={'-m': ModelArg('brain-tumor-segmentation-0002'), + '-o': '.', + '-ms': '1,2,3,0'}), single_option_cases('-i', *DATA_SEQUENCES['brain-tumor-nifti']), )), @@ -1193,7 +1194,6 @@ def single_option_cases(key, *args): *combine_cases( TestCase(options={'--architecture_type': 'yolo'}), single_option_cases('-m', - ModelArg('mobilefacedet-v1-mxnet'), ModelArg('mobilenet-yolo-v4-syg'), ModelArg('person-vehicle-bike-detection-crossroad-yolov3-1020'), ModelArg('yolo-v1-tiny-tf'),