onnx轉成nef格式出現KeyError: 'images'的問題
如附圖,使用yolox生出pth轉成onnx模型,之後對onnx模型使用editor.py以及onnx2onnx.py
轉換過程卻出現KeyError: 'images',想請問是什麼問題,謝謝
The discussion has been closed due to inactivity. To continue with the topic, please feel free to post a new discussion.
Comments
您好,這部分我已經解決了
想請問耐能教育中心提供的YOLOX範例demo,標記出來的結果圖要如何產生標籤與預測結果呢? 如下圖範例
@蔡泊均
Hi 泊均,
如果您是使用耐能教育中心提供的 YOLOX 範例 demo,您想要結果圖要產生標籤與預測結果您可以從 KL720DemoGenericInferenceYoloX_BypassHwPreProc.py 這個 scripts 內的 dets 取得 標籤 (label) 和 預測結果 (score) 資訊,取得標籤和預測結果資訊後,您就可以根據您的 label 與 score 畫在推論結果圖上。
dets = yolox_postprocess(prediction_np, num_classes, prob_threshold, iou_threshold, topk)
Returns:
dets: (N, 6): The detected objects, The last dimension is [x1, y1, x2, y2, score, label]
了解,謝謝您的回覆