object_detection/user_config.yaml

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}