PyTorch C++ API を使って Resize したい
torchvisionのResizeを、OpenCV および PyTorch C++ API を使ってなるべく再現します。ここで紹介するのは、あくまでも「同じような操作」を行う方法なので、torchvisionと全く同じ前処理ができる訳ではないです。(PyTorch Forum)Pytho
2022年 4月の記事一覧
記事数
4
PyTorch C++ APIを使ってcenterCropをしたい
torchvisionで実装されているcenterCropをC++(OpenCV)で再現する方法です。torchvisionで使われているcenterCropの実装を見れば終わりなんですが、そこまでしたくない人のために実装例を載せます。cv::Mat img_float; // c
PyTorchモデルがTorchScriptに変換できないときに確認すること
torch.jit.traceを試してみるtorch.jit.script(model)がうまくいかない場合でも、torch.jit.trace(model, sample_input)ならうまくいく場合があります。torch.jit.traceを使った場合、変換後
C++でOpenCVを使った時にハマったこと
PyTorchと併用した際の undefined reference to cv::imreadundefined referenceは、リンクすべきライブラリをリンクし忘れている場合に発生するエラーですが、今回はきちんとリンクしているのにエラーが消えませんでした。自分の場合は、P