RLL Computer Vision Code: code/armour_detector.h 源文件

RLL Computer Vision

RLL Computer Vision Code 1.0
江苏科技大学HLL战队机器视觉代码
code
 /*****************************************************************************
 * HLL Computer Vision Code *
 * Copyright (C) 2017 HLL [email protected]. *
 * *
 * This file is part of HCVC. *
 * *
 * @file armour_detector.h *
 * @brief Detect the armour zone *
 * Details. *
 * *
 * @author HLL *
 * @email [email protected] *
 * @version 1.0.0.0 *
 * @date 2017.11.10 *
 * *
 *----------------------------------------------------------------------------*
 * Remark : Description *
 *----------------------------------------------------------------------------*
 * Change History : *
 * <Date> | <Version> | <Author> | <Description> *
 *----------------------------------------------------------------------------*
 * 2017/11/10 | 1.0.0.0 | Zhu Min | Create file *
 *----------------------------------------------------------------------------*
 * *
 *****************************************************************************/
 
 #ifndef ARMOUR_DETECTOR_H
 #define ARMOUR_DETECTOR_H
 
 #include "image_preprocessor.h"
 
 namespace HCVC {
 
 class ArmourDetector: public ImagePreprocessor
 {
 public:
  ArmourDetector();
 
  bool detect(const Mat& srcImage);
 
  Rect2d getBestArmourBlock() const;
 
  void drawBlocks(Mat srcImage, const vector<RotatedRect>& minRotatedRects, const Scalar& color) const;
 
  struct Params
  {
  float angleRange; 
  float minArea; 
  float maxHeightWidthRat; 
  float minHeightWidthRat; 
  float maxWidthRat; 
  float maxHeightRat; 
  float maxAngleDiff; 
  float maxHeightGapRat; 
  float minHeightGapRat; 
  }params;
 
 private:
  struct OptimalArmourBlock
  {
  RotatedRect block;
  double grade;
 
  OptimalArmourBlock(const RotatedRect& _block, const double& _grade):
  block(_block), grade(_grade)
  {}
 
  bool operator < (const OptimalArmourBlock& other) const
  {
  return grade < other.grade;
  }
  };
 
  //！ 按顺序存储最优的装甲板区域
  vector<OptimalArmourBlock> optimalArmourBlocks;
 
  vector<vector<Point> > searchBlocks(Mat srcImage);
 
  void fillLampBlock(Mat& srcImage, vector<vector<Point> >& blocks, int row, int col);
 
  vector<RotatedRect> calcBlocksInfo(const vector<vector<Point> >& blocks);
 
  vector<RotatedRect> extracArmourBlocks(const vector<RotatedRect>& minRotatedRects);
 
  void markArmourBlocks(const Mat& srcImage, const Mat& dstImage, const vector<RotatedRect>& armourBlocks);
 
  void cutEdgeOfRect(Point2f* points);
 };
 }
 #endif // ARMOUR_DETECTOR_H
制作者
1.8.13
previous page start next page