You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
600 B
24 lines
600 B
9 months ago
|
#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);
|