FBAGSTM's picture
STM32 AI Experimentation Hub
747451d
operation_mode: prediction
model:
model_path: /app/object_detection/pretrained_models/st_yoloxn_d033_w025_320_int8.tflite
model_type: st_yoloxn
input_shape: (320,320,3)
dataset:
format: tfs
dataset_name: coco
class_names: [ person ]
prediction_path: /app/object_detection/datasets
preprocessing:
rescaling: { scale: 1/127.5, offset: -1 }
resizing:
aspect_ratio: fit
interpolation: nearest
color_mode: rgb
postprocessing:
confidence_thresh: 0.5
NMS_thresh: 0.5
IoU_eval_thresh: 0.5
plot_metrics: True
max_detection_boxes: 10
mlflow:
uri: ./tf/src/experiments_outputs/mlruns
hydra:
run:
dir: ./tf/src/experiments_outputs/${now:%Y_%m_%d_%H_%M_%S}