class PersonDetection0200NNCFint8( create_object_detection_nncf_test_case( 'person-detection', 'person-detection-0200', os.path.dirname(__file__) + '/../../../data/airport/annotation_person_train.json', os.path.dirname(__file__) + '/../../../data/airport/train', '--nncf-quantization')): """ Test case for person-detection-0200 model with NNCF int8 compression. """
class FaceDetection0202NNCFint8( create_object_detection_nncf_test_case( 'face-detection', 'face-detection-0202', os.path.dirname(__file__) + '/../../../data/airport/annotation_faces_train.json', os.path.dirname(__file__) + '/../../../data/airport/', '--nncf-quantization')): """ Test case for face-detection-0202 model with NNCF int8 compression. """
class FaceDetection0200NNCFint8FromTemplate( create_object_detection_nncf_test_case( 'face-detection', 'face-detection-0200', os.path.dirname(__file__) + '/../../../data/airport/annotation_faces_train.json', os.path.dirname(__file__) + '/../../../data/airport/', '', {'optimisations.nncf_quantization.default': 1})): """ Test case for face-detection-0200 model with NNCF int8 compression initialized from template.yaml. """
class VehicleDetection0202NNCFint8( create_object_detection_nncf_test_case( 'vehicle-detection', 'vehicle-detection-0202', os.path.dirname(__file__) + '/../../../data/vehicle_detection/annotation_train.json', os.path.dirname(__file__) + '/../../../data/vehicle_detection/train', '--nncf-quantization')): """ Test case for vehicle-detection-0202 model with NNCF int8 compression. """
class FaceDetection0200NNCFint8sparsity( create_object_detection_nncf_test_case( 'face-detection', 'face-detection-0200', os.path.dirname(__file__) + '/../../../data/airport/annotation_faces_train.json', os.path.dirname(__file__) + '/../../../data/airport/', '--nncf-quantization --nncf-sparsity', {}, {'nncf_sparsity.total_epochs': 4})): """ Test case for face-detection-0200 model with NNCF int8 and sparsity compression. """
class PersonVehicleBikeDetection2002NNCFint8( create_object_detection_nncf_test_case( 'person-vehicle-bike-detection', 'person-vehicle-bike-detection-2002', os.path.dirname(__file__) + '/../../../data/airport/annotation_example_train.json', os.path.dirname(__file__) + '/../../../data/airport/train', '--nncf-quantization')): """ Test case for person-vehicle-bike-detection-2002 model with NNCF int8 compression. """
class FaceDetection0204NNCFint8( create_object_detection_nncf_test_case( 'face-detection', 'face-detection-0204', os.path.dirname(__file__) + '/../../../data/airport/annotation_faces_train.json', os.path.dirname(__file__) + '/../../../data/airport/', '--nncf-quantization', test_export_threshold=0.21 # TODO(lbeynens): it's very big threshold, try to fix this ) ): """ Test case for face-detection-0204 model with NNCF int8 compression. """