CPUのみでAIの学習をするシリーズ第8弾。
yolo_v8のcpp版を作ってみました。
学習も再学習も推論もフルでCPUだけできます。
世界初。これ作るの夢だったんだよね。
最初claudeに言ったら、これを作るのはとても大変ですと言っていたんですが、libtorchにtorchscript読み込ませて、各関数の自動微分とバックプロパゲーションの結果を一致させれば作れるだろって言ったら納得して作ってくれました。
yoloってきちんとソースコードを整理するとcoreの部分なんてほんのちょっとしかないんですよ。
https://github.com/yomei-o/yolov8_cpp
0 件のコメント:
コメントを投稿