You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
713 B
C
30 lines
713 B
C
5 years ago
|
//
|
||
|
|
// Created by Jack Yu on 22/09/2017.
|
||
|
|
//
|
||
|
|
|
||
|
|
#ifndef HYPERPR_FINEMAPPING_H
|
||
|
|
#define HYPERPR_FINEMAPPING_H
|
||
|
|
|
||
|
|
#include <opencv2/dnn.hpp>
|
||
|
|
#include <opencv2/opencv.hpp>
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
namespace pr {
|
||
|
|
class FineMapping {
|
||
|
|
public:
|
||
|
|
FineMapping();
|
||
|
|
|
||
|
|
FineMapping(std::string prototxt, std::string caffemodel);
|
||
|
|
static cv::Mat FineMappingVertical(cv::Mat InputProposal, int sliceNum = 15,
|
||
|
|
int upper = 0, int lower = -50,
|
||
|
|
int windows_size = 17);
|
||
|
|
cv::Mat FineMappingHorizon(cv::Mat FinedVertical, int leftPadding,
|
||
|
|
int rightPadding);
|
||
|
|
|
||
|
|
private:
|
||
|
|
cv::dnn::Net net;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace pr
|
||
|
|
#endif // HYPERPR_FINEMAPPING_H
|