برای هر ردیف که در واقع خروجی یک inference هستش باید max بدست بیارید محل میشه شناسه کلاس و مقدارش میشه confidence :
void getMaxClass(const cv::Mat & probBlob, int * class_index, double * confidence)
{
cv::Mat probMat = probBlob.reshape(1, 1);
cv::Point class_number;
minMaxLoc(probMat, NULL, confidence, NULL, &class_number);
*class_index = class_number.x;
}
یا
y = np.argmax(res, axis=-1)