2022-09-01から1ヶ月間の記事一覧

kotlinのonnxでmmsegmentationの推論

・ネットワーク選定onnxでAdaptive Poolingが使えないので、Adaptive Poolingを使ってないネットワークを選定する。mmsegmentationのデモで使われていpspnetはダメ。fcnは大丈夫。・onnxモデルへの変換直接onnxモデルへ変換しようとするとエラーが出る。以下…

kotlinのBufferedImageからRGB値を抽出

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) }}

kotlinでonnx推論

以下のリポジトリを分かりやすく整理しました。onnxruntime/ScoreMNIST.java at main · microsoft/onnxruntime · GitHub データはこちらからダウンロード。 LIBSVM Data: Classification (Multi Class) onnxモデルは以下の手順で生成。 onnxの使い方 - LeMU_…

loss, accuracyの計算方法まとめ

criterionのreductionはデフォルトで'mean'。len(data_loader)はバッチ数。なので、epochの平均lossの算出は、バッチごとの(平均)lossを加算していって、len(data_loader)で割ればよい。正解数は correct += pred.eq(target.view_as(pred)).sum().item() …

onnxの使い方

・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)) …

call by value, reference, sharing

pythonの参照について - Qiita 値渡しと参照渡しの違いを理解する

mmsegmentation使い方

・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・自前デ…