FBAGSTM's picture
STM32 AI Experimentation Hub
747451d
operation_mode: quantization
model:
model_path: ../../stm32ai-modelzoo/object_detection/yolov2t/ST_pretrainedmodel_public_dataset/coco_2017_person/yolov2t_224/yolov2t_224.keras
model_type: yolov2t
dataset:
format: tfs
dataset_name: coco
class_names: [ person ]
quantization_path: /local/data/od_validation_dataset_coco_person/
preprocessing:
rescaling: { scale: 1/127.5, offset: -1 }
resizing:
aspect_ratio: fit
interpolation: nearest
color_mode: rgb
postprocessing:
confidence_thresh: 0.001
NMS_thresh: 0.5
IoU_eval_thresh: 0.5
plot_metrics: False # Plot precision versus recall curves. Default is False.
max_detection_boxes: 100
quantization:
quantizer: TFlite_converter
quantization_type: PTQ
quantization_input_type: uint8
quantization_output_type: float
export_dir: quantized_models
mlflow:
uri: ./tf/src/experiments_outputs/mlruns
hydra:
run:
dir: ./tf/src/experiments_outputs/${now:%Y_%m_%d_%H_%M_%S}