Custom Object Detection model with MLFlow tracking

[diegosaenzmaqta]

I am trying to get the Yolov8 object detection model as a MLFlow python function to then automatically score and evaluate from the saved model. When I am setting mlflow_extension.set_run_inference_info and project.create_mlflow_pyfunc_model to "OTHER" prediction type, I cant use the mlflow_version.evaluate method on the selected saved model.

Any ideas how to set up the prediction type for this task?

---

Operating system used: Oracle Linux Server 8.8

---

Operating system used: Oracle Linux Server 8.8

[HarizoR]

Hi,

Unfortunately Dataiku does not have the ability to evaluate object detection models in its mlflow integration. You can use it for scoring, but not for evaluation reports.

Best,

Harizo