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. """
