تبدیل خروجی مدل به مقدار یا شناسه کلاس

Question

وقتی من مدل ترین شده ای را در حالت predict تست می کنم خروجی مدل به صورت مقادیر زیر نمایش داده میشه آیا نباید خروجی به صورت یک مقدار عددی که نشان دهنده برچسب کلاس مورد نظر باشه یا سوال رو جور دیگه ای بپرسم امکان داره که این خروجی را به لیبل تبدیل کنم؟

[[0.8544417  0.14555828]
 [0.9312985  0.06870154]]

 

Answer 1

برای هر ردیف که در واقع خروجی یک 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)