def test_object_detection(self):
json_dict = fetch(topology_types=['object_detection'])
assert validate(json_dict) == True
check_topology_types(json_dict, ['object_detection'])
check_adapters(json_dict,
['ssd', 'tiny_yolo_v1', 'yolo_v2', 'yolo_v3'])
check_converters(json_dict, ['voc_detection'])
def test_all_parameters(self):
json_dict = fetch()
assert validate(json_dict) == True
assert 'datasets' in json_dict.keys()
dataset_content = json_dict['datasets'].keys()
assert 'adapter' in dataset_content
assert 'converter' in dataset_content
assert 'metric' in dataset_content
assert 'preprocessor' in dataset_content
assert 'postprocessor' in dataset_content
assert 'models' in json_dict.keys()
models_content = json_dict['models'].keys()
assert 'launcher' in models_content
assert 'topology_type' in json_dict.keys()
def test_yolo_v1_tiny(self):
json_dict = fetch(topology_types=['yolo_v1_tiny'])
assert validate(json_dict) == True
check_topology_types(json_dict, ['yolo_v1_tiny'])
check_adapters(json_dict, ['tiny_yolo_v1'])
check_converters(json_dict, ['voc_detection'])
def test_image_classification(self):
json_dict = fetch(topology_types=['image_classification'])
assert validate(json_dict) == True
check_topology_types(json_dict, ['image_classification'])
check_adapters(json_dict, ['classification'])
check_converters(json_dict, ['imagenet'])
def test_launchers(self):
json_dict = fetch(launchers=['dlsdk', 'opencv'])
assert validate(json_dict) == True
check_launchers(json_dict, ['dlsdk', 'opencv'])
def test_faster_rcnn(self):
json_dict = fetch(topology_types=['faster_rcnn'])
assert validate(json_dict) == True
check_topology_types(json_dict, ['faster_rcnn'])
check_adapters(json_dict, ['ssd'])
check_converters(json_dict, ['voc_detection'])