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'])