为Rikai提供Yolov5支持

rikai-yolov5集成了基于PyTorch实现的Yolov5，并整合到Rikai中。它基于打包的ultralytics/yolov5。

笔记本

• 使用Rikai分析周杰伦的《 Mojito》中的图片。

使用方法

使用rikai-yolov5有两种方式。

rikai.mlflow.pytorch.log_model(
    model,
    "model",
    OUTPUT_SCHEMA,
    registered_model_name=registered_model_name,
    model_type="yolov5",
)

另一种方式是在Rikai SQL中设置model_type

CREATE MODEL mlflow_yolov5_m
MODEL_TYPE yolov5
OPTIONS (
  device='cpu'
)
USING 'mlflow:///{registered_model_name}';

可用选项

以下是上述选项的示例用法

CREATE MODEL mlflow_yolov5_m
OPTIONS (
  device='cpu',
  iou_thres=0.5
)
USING 'mlflow:///{registered_model_name}';