DockerでGPU版ONNXを使ってみる
1.7k{icon} {views} DockerでGPU版をONNXを動かしてみました。比較的に簡単に動かせたので、いろいろ便利だと思います。YOLOXを例に検証します。 概要 ONNXRuntimeのGPU版をDoc […]...
訓練済みYOLOXでサクッと推論したい場合に使える方法
5.9k{icon} {views} 物体検出を使いたい場合、カスタムデータで訓練することなく、COCO Pretrainedモデルによる推論だけで事足りることも多々あります。2021年最新モデルのYOLOXを使って、サ […]...
PyTorch→ONNXのコンバートでモデルの入力サイズを可変にして推論する
8.4k{icon} {views} ONNXでモデルは入力サイズを可変にできます。PyTorch→ONNXの変換と、ONNXRuntimeでの推論方法、また可変にしたことによる速度の副作用を検証していきます。 きっかけ […]...