Init commit
This commit is contained in:
23
vs_linux/include/prior_boxes.hpp
Normal file
23
vs_linux/include/prior_boxes.hpp
Normal file
@ -0,0 +1,23 @@
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <cmath>
|
||||
|
||||
class PriorBox {
|
||||
|
||||
|
||||
private:
|
||||
std::vector<std::vector<int>> feature_maps;
|
||||
std::vector<std::vector<float>> min_sizes;
|
||||
std::vector<int> steps;
|
||||
bool clip;
|
||||
std::vector<int> image_size;
|
||||
std::string name;
|
||||
|
||||
|
||||
public:
|
||||
PriorBox(std::vector<int> image_size = std::vector<int>(), std::string phase = "train");
|
||||
std::vector<std::vector<float>> forward();
|
||||
|
||||
};
|
||||
|
||||
std::vector<std::vector<float>> decode(const std::vector<std::vector<float>>& loc, const std::vector<std::vector<float>>& priors, const std::vector<float>& variances);
|
12
vs_linux/include/reader.hpp
Normal file
12
vs_linux/include/reader.hpp
Normal file
@ -0,0 +1,12 @@
|
||||
#pragma once
|
||||
|
||||
#include <iostream>
|
||||
#include <fstream>
|
||||
#include <sstream>
|
||||
#include <vector>
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
vector<vector<float>> splitFloats(const vector<float>& floats, int index) ;
|
||||
vector<float> readFloatsFromFile(const string& filename);
|
6
vs_linux/include/utils.hpp
Normal file
6
vs_linux/include/utils.hpp
Normal file
@ -0,0 +1,6 @@
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <cmath>
|
||||
#include <algorithm>
|
||||
|
||||
std::vector<std::vector<float>> nms(std::vector<std::vector<float>>& bboxes, float threshold);
|
Reference in New Issue
Block a user