Yolo_Label/label_img.h

97 lines
2.2 KiB
C
Raw Normal View History

2018-10-29 07:26:06 +03:00
#ifndef LABEL_IMG_H
#define LABEL_IMG_H
#include <QObject>
#include <QLabel>
#include <QImage>
#include <QMouseEvent>
#include <iostream>
#include <fstream>
struct ObjectLabelingBox
{
int label;
QRectF box;
2018-10-29 07:26:06 +03:00
};
class label_img : public QLabel
{
Q_OBJECT
public:
label_img(QWidget *parent = nullptr);
void mouseMoveEvent(QMouseEvent *ev);
void mousePressEvent(QMouseEvent *ev);
void mouseReleaseEvent(QMouseEvent *ev);
QVector<QColor> m_drawObjectBoxColor;
QStringList m_objList;
2018-10-29 07:26:06 +03:00
int m_uiX;
int m_uiY;
int m_imgX;
int m_imgY;
bool m_bLabelingStarted;
bool m_bVisualizeClassName;
2018-10-29 07:26:06 +03:00
static QColor BOX_COLORS[10];
QVector <ObjectLabelingBox> m_objBoundingBoxes;
void init();
2018-12-14 07:44:18 +03:00
void openImage(const QString &, bool& ret);
2018-10-29 07:26:06 +03:00
void showImage();
void loadLabelData(const QString & );
void setFocusObjectLabel(int);
void setFocusObjectName(QString);
2021-11-29 18:04:33 +03:00
void setContrastGamma(float);
2018-12-14 07:44:18 +03:00
bool isOpened();
QImage crop(QRect);
QRectF getRelativeRectFromTwoPoints(QPointF , QPointF);
QRect cvtRelativeToAbsoluteRectInUi(QRectF);
QRect cvtRelativeToAbsoluteRectInImage(QRectF);
QPoint cvtRelativeToAbsolutePoint(QPointF);
QPointF cvtAbsoluteToRelativePoint(QPoint);
2021-11-29 18:04:33 +03:00
2018-10-29 07:26:06 +03:00
signals:
void Mouse_Moved();
void Mouse_Pressed();
void Mouse_Release();
private:
int m_focusedObjectLabel;
QString m_foucsedObjectName;
double m_aspectRatioWidth;
double m_aspectRatioHeight;
QImage m_inputImg;
2021-11-29 18:04:33 +03:00
QImage m_resized_inputImg;
QPointF m_relative_mouse_pos_in_ui;
QPointF m_relatvie_mouse_pos_LBtnClicked_in_ui;
2018-10-29 07:26:06 +03:00
2021-11-29 18:04:33 +03:00
unsigned char m_gammatransform_lut[256];
QVector<QRgb> colorTable;
2018-10-29 07:26:06 +03:00
void setMousePosition(int , int);
void drawCrossLine(QPainter& , QColor , int thickWidth = 3);
void drawFocusedObjectBox(QPainter& , Qt::GlobalColor , int thickWidth = 3);
void drawObjectBoxes(QPainter& , int thickWidth = 3);
void drawObjectLabels(QPainter& , int thickWidth = 3, int fontPixelSize = 14, int xMargin = 5, int yMargin = 2);
2021-11-29 18:04:33 +03:00
void gammaTransform(QImage& image);
void removeFocusedObjectBox(QPointF);
2018-10-29 07:26:06 +03:00
};
#endif // LABEL_IMG_H