2022-09-01から1ヶ月間の記事一覧
・ネットワーク選定onnxでAdaptive Poolingが使えないので、Adaptive Poolingを使ってないネットワークを選定する。mmsegmentationのデモで使われていpspnetはダメ。fcnは大丈夫。・onnxモデルへの変換直接onnxモデルへ変換しようとするとエラーが出る。以下…
val img = ImageIO.read(File("test.png"))for (y in 0 until img.height) { for (x in 0 until img.width) { val color = Color(img.getRGB(x, y)) val r = color.red val g = color.green val b = color.blue // img.setRGB(x, y, color.rgb) }}
以下のリポジトリを分かりやすく整理しました。onnxruntime/ScoreMNIST.java at main · microsoft/onnxruntime · GitHub データはこちらからダウンロード。 LIBSVM Data: Classification (Multi Class) onnxモデルは以下の手順で生成。 onnxの使い方 - LeMU_…
criterionのreductionはデフォルトで'mean'。len(data_loader)はバッチ数。なので、epochの平均lossの算出は、バッチごとの(平均)lossを加算していって、len(data_loader)で割ればよい。正解数は correct += pred.eq(target.view_as(pred)).sum().item() …
・pthファイルからonnxファイルへの変換 import torchfrom torch import onnxfrom torchvision import transforms, datasetsdef main(): transform = transforms.Compose([ transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) …
pythonの参照について - Qiita 値渡しと参照渡しの違いを理解する
・git cloneGitHub - open-mmlab/mmsegmentation: OpenMMLab Semantic Segmentation Toolbox and Benchmark.・get_started.md step0にmim install mmengineを追加。 mmsegmentation/get_started.md at master · open-mmlab/mmsegmentation · GitHub・自前デ…