class ExportTestCase(export_test_case('face-detection', model_name, alt_ssd_export=alt_ssd_export)):
def setUp(self):
super().setUp()
assert replace_text_in_file(self.configuration_file, 'data/WIDERFace',
'../../data/airport')
assert replace_text_in_file(self.configuration_file, 'val.json',
'annotation_faces_train.json')
class ExportTestCase(
export_test_case('horizontal-text-detection', model_name)):
def setUp(self):
super().setUp()
assert replace_text_in_file(
self.configuration_file, 'data/text-dataset/',
'../../data/horizontal_text_detection/')
assert replace_text_in_file(self.configuration_file,
'IC13TEST.json', 'annotation.json')
class PersonDetection0202TestCase(
export_test_case('person-detection',
'person-detection-0202',
'person-detection-0202-1.pth',
alt_ssd_export=True)):
""" Test case for person-detection-0202 export. """
class PersonVehicleBikeDetection2002TestCase(
export_test_case('person-vehicle-bike-detection',
'person-vehicle-bike-detection-2002',
'vehicle-person-bike-detection-2002-1.pth',
alt_ssd_export=True)):
""" Test case for person-vehicle-bike-detection-2002 export. """
class PersonVehicleBikeDetectionCrossroad1016TestCase(
export_test_case(
'person-vehicle-bike-detection-crossroad-1016',
'person_vehicle_bike_sd512_mb2_clustered_epoch_21.pth', True)):
""" Test case for person-vehicle-bike-detection-crossroad-1016 export. """