1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
string modelCfg;
string modelBin;
string classFilename;

if (idx == 0){

	modelCfg = "yolo/yolov2-tiny.cfg";
	modelBin = "yolo/yolov2-tiny.weights";
	classFilename = "yolo/darknet-master/data/coco.names";
}
if (idx == 1){

	modelCfg = "yolo/yolov2.cfg";
	modelBin = "yolo/yolov2.weights";
	classFilename = "yolo/darknet-master/data/coco.names";
}
if (idx == 2){

	modelCfg = "yolo/yolov2-voc.cfg";
	modelBin = "yolo/yolov2-tiny-voc.weights";
	classFilename = "yolo/darknet-master/data/voc.names";
}
try{

	net = dnn::readNetFromDarknet(modelCfg, modelBin);
}catch (cv::Exception &ee){

	QMessageBox::warning(this, "Exception", ee.what());
	if (net.empty()){

		QMessageBox::warning(this, "Exception", "Can't load the network by using the flowing files.");
		return;
	}
}

// Load Label
classNamesVec.clear();
ifstream classNamesFile(classFilename);
if (classNamesFile.is_open()){

	string className = "";
	while (std::getline(classNamesFile, className))
		classNamesVec.push_back(className);
}
else{

	QMessageBox::warning(this, "Exception", "Load label failed.");
	return;
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
Mat frame;
image.copyTo(frame);
if (frame.empty()){

	QMessageBox::warning(this, "Warning", "image is empty, please check!");
	return;
}
if (frame.channels() == 4) cvtColor(frame, frame, COLOR_BGRA2BGR);

double ttt = (double)cvGetTickCount();

Mat inputBlob = blobFromImage(frame, 1 / 255.F, Size(416, 416), Scalar(), true, false); //Convert Mat to batch of images
net.setInput(inputBlob, "data");                   //set the network input]
Mat detectionMat = net.forward("detection_out");   //compute output

ttt = (double)cvGetTickCount() - ttt;
ui.labelTime->setText(toChinese("识别时间：") + QString::number(ttt / (cvGetTickFrequency() * 1000000)) + toChinese("秒"));

float confidenceThreshold = ui.dsbConfidence->value();
for (int i = 0; i < detectionMat.rows; i++){

	const int probability_index = 5;
	const int probability_size = detectionMat.cols - probability_index;
	float *prob_array_ptr = &detectionMat.at<float>(i, probability_index);

	size_t objectClass = max_element(prob_array_ptr, prob_array_ptr + probability_size) - prob_array_ptr;
	float confidence = detectionMat.at<float>(i, (int)objectClass + probability_index);

	if (confidence > confidenceThreshold){

		float x_center = detectionMat.at<float>(i, 0) * frame.cols;
		float y_center = detectionMat.at<float>(i, 1) * frame.rows;
		float width = detectionMat.at<float>(i, 2) * frame.cols;
		float height = detectionMat.at<float>(i, 3) * frame.rows;
		Point p1(cvRound(x_center - width / 2), cvRound(y_center - height / 2));
		Point p2(cvRound(x_center + width / 2), cvRound(y_center + height / 2));
		Rect object(p1, p2);

		Scalar object_roi_color(0, 255, 0);

		if (1)//
		{
			rectangle(frame, object, object_roi_color);
		}
		else
		{
			Point p_center(cvRound(x_center), cvRound(y_center));
			line(frame, object.tl(), p_center, object_roi_color, 1);
		}

		String className = objectClass < classNamesVec.size() ? classNamesVec[objectClass] : cv::format("unknown(%d)", objectClass);
		String label = format("%s: %.2f", className.c_str(), confidence);
		int baseLine = 0;
		Size labelSize = getTextSize(label, FONT_HERSHEY_SIMPLEX, 0.5, 1, &baseLine);
		rectangle(frame, Rect(p1, Size(labelSize.width, labelSize.height + baseLine)), object_roi_color, FILLED);
		putText(frame, label, p1 + Point(0, labelSize.height), FONT_HERSHEY_SIMPLEX, 0.5, Scalar(0, 0, 0));
	}
}