amakazeryu https://blog.hatena.ne.jp/amakazeryu/ とある大学生の勉強メモ https://amakazeryu.hatenablog.com/ 推論を行ったとき結果がnullとなるケースが多々ある． 今回PreTrainedのresnetをonnx exportし，onnx runtimeで推論すると結果が返るが， pytorchで学習を進めたモデルをexportした場合，推論結果がnullとなった際の備忘録． 結論 画像を配列にしたとき正規化を行えば推論結果が返るようになった． input_image = cv2.imread(file_path)/255 序論 推論時はモデルの入出力の型に注意しなくてはいけない． 今回画像をTensorFloat配列に変換しモデルに入力する． UWPにonnxモデルを取り込むとcode gener… 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Famakazeryu.hatenablog.com%2Fentry%2F2022%2F01%2F15%2F115918" title="学習したモデルの推論結果がnullになる　MaskRCNN(Pytorch)，onnxruntime - とある大学生の勉強メモ" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> https://cdn-ak.f.st-hatena.com/images/fotolife/a/amakazeryu/20220115/20220115115359.png Hatena Blog https://hatena.blog 2022-01-15 11:59:18 rich https://amakazeryu.hatenablog.com/entry/2022/01/15/115918 1.0 100%