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

RLL Computer Vision

previous page next page
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