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