diff --git a/label_img.cpp b/label_img.cpp index e968c22..119178d 100644 --- a/label_img.cpp +++ b/label_img.cpp @@ -269,13 +269,19 @@ void label_img::drawObjectBoxes(QPainter& painter, int thickWidth) { QPen pen; pen.setWidth(thickWidth); + QFont font = painter.font(); + font.setPixelSize(12); + font.setBold(true); + painter.setFont(font); for(ObjectLabelingBox boundingbox: m_objBoundingBoxes) { pen.setColor(m_drawObjectBoxColor.at(boundingbox.label)); painter.setPen(pen); - painter.drawRect(cvtRelativeToAbsoluteRectInUi(boundingbox.box)); + QRect rectUi = cvtRelativeToAbsoluteRectInUi(boundingbox.box); + painter.drawRect(rectUi); + painter.drawText(rectUi.topLeft() + QPoint(5, 14 + 14 * boundingbox.label), m_objList.at(boundingbox.label)); } } diff --git a/label_img.h b/label_img.h index 1badeda..b862217 100644 --- a/label_img.h +++ b/label_img.h @@ -25,6 +25,7 @@ public: void mouseReleaseEvent(QMouseEvent *ev); QVector m_drawObjectBoxColor; + QStringList m_objList; int m_uiX; int m_uiY; diff --git a/mainwindow.cpp b/mainwindow.cpp index 4c867bf..4b5dc37 100644 --- a/mainwindow.cpp +++ b/mainwindow.cpp @@ -250,6 +250,7 @@ void MainWindow::load_label_list_data(QString qstrLabelListFile) ui->label_image->m_drawObjectBoxColor.push_back(labelColor); } + ui->label_image->m_objList = m_objList; } }