Pytorchでは正しい出力なのに、変換後がおかしい
カスタムデータでトレーニングしたYolov8を変換して、previewも表示されるのに、いざSwiftから使おうとするとメチャクチャな結果が出ることがありました。
バグ
iOS17.0
MacOS Sonoma
の時にこのバグは発生しました。
MLE5Engineというやつのバグらしい。
対処法
以下で正しい出力になりました。
var config = MLModelConfiguration()
config.setValue(1, forKey: "experimentalMLE5EngineUsage")
let model = try custom_yolo_coreml_model(configuration: config).model
🐣
フリーランスエンジニアです。
AIについて色々記事を書いていますのでよかったらプロフィールを見てみてください。
もし以下のようなご要望をお持ちでしたらお気軽にご相談ください。
AIサービスを開発したい、ビジネスにAIを組み込んで効率化したい、AIを使ったスマホアプリを開発したい、
ARを使ったアプリケーションを作りたい、スマホアプリを作りたいけどどこに相談したらいいかわからない…
いずれも中間コストを省いたリーズナブルな価格でお請けできます。
お仕事のご相談はこちらまで
rockyshikoku@gmail.com
機械学習やAR技術を使ったアプリケーションを作っています。
機械学習／AR関連の情報を発信しています。