ゆるエンジニアはいろいろ遊びたい

FAエンジニアが週末にいろいろ遊ぶブログです

Python

YOLOをGPUで動かしてみた

ShelPiで使っている物体検出モデルのYOLOですが、現在CPUで動作しています。 これでも全然高速なのですが、今回はGPUで動かしてみたいと思います。 GPUで動作させるためには、pytorchというものを入れる必要があるようです。 pip install torch torchvision …