Skip to content

Instantly share code, notes, and snippets.

@pliablepixels
Last active December 25, 2020 12:05
Show Gist options
  • Save pliablepixels/f6a38ca699b6dc2b309ec5edb96831f6 to your computer and use it in GitHub Desktop.
Save pliablepixels/f6a38ca699b6dc2b309ec5edb96831f6 to your computer and use it in GitHub Desktop.
sample_mlsequence.ini
use_sequence = yes
ml_sequence= {
'general': {
'model_sequence': 'object,face,alpr'
},
'object': {
'general':{
'pattern':'(person)',
'same_model_sequence_strategy': 'first' # also 'most', 'most_unique's
},
'sequence': [{
#First run on TPU with higher confidence
'object_weights':'{{base_data_path}}/models/coral_edgetpu/ssd_mobilenet_v2_coco_quant_postprocess_edgetpu.tflite',
'object_labels': '{{base_data_path}}/models/coral_edgetpu/coco_indexed.names',
'object_min_confidence': 0.6,
'object_framework':'coral_edgetpu'
},
{
# YoloV4 on GPU if TPU fails (because sequence strategy is 'first')
'object_config':'{{base_data_path}}/models/yolov4/yolov4.cfg',
'object_weights':'{{base_data_path}}/models/yolov4/yolov4.weights',
'object_labels': '{{base_data_path}}/models/yolov4/coco.names',
'object_min_confidence': 0.3,
'object_framework':'opencv',
'object_processor': 'gpu'
}]
},
'face': {
'general':{
'pattern': '.*',
'same_model_sequence_strategy': 'first'
},
'sequence': [{
'face_detection_framework': 'dlib',
'known_images_path': '{{base_data_path}}/known_faces',
'face_model': 'cnn',
'face_train_model': 'cnn',
'face_recog_dist_threshold': 0.6,
'face_num_jitters': 1,
'face_upsample_times':1
}]
},
'alpr': {
'general':{
'same_model_sequence_strategy': 'first',
'pre_existing_labels':['car', 'motorbike', 'bus', 'truck', 'boat'],
},
'sequence': [{
'alpr_api_type': 'cloud',
'alpr_service': 'plate_recognizer',
'alpr_key': '!PLATEREC_ALPR_KEY',
'platrec_stats': 'no',
'platerec_min_dscore': 0.1,
'platerec_min_score': 0.2,
}]
}
}
~
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment