AutoCar  v1.0.0
labeler.h
1 #ifndef LABELER_H
2 #define LABELER_H
3 
4 #include <iostream>
5 #include <stdio.h>
6 #include <vector>
7 #include <opencv2/opencv.hpp>
8 #include "armor_detect.h"
9 
10 namespace autocar {
11 namespace vision_mul {
12  class labeler {
13  public:
14  labeler(std::string img_dir_, std::string label_img_dir_, std::string label_txt_dir_, int cols_=640, int rows_=480);
15  ~labeler();
16  void gen_sample(const cv::Mat &image, std::vector<armor_info> armors);
17 
18  private:
19  std::string img_dir;
20  std::string label_img_dir;
21  std::string label_txt_dir;
22  int rows;
23  int cols;
24  int count;
25  };
26 }
27 }
28 
29 #endif // LABELER_H
Definition: vel2odom.h:5
Definition: labeler.h:12