Modifikace README, zadání 1.0
This commit is contained in:
29
vs_win/include/prior_boxes.hpp
Normal file
29
vs_win/include/prior_boxes.hpp
Normal file
@ -0,0 +1,29 @@
|
||||
/*
|
||||
* Prior boxes for RetinaNet
|
||||
* Tomas Goldmann,2023
|
||||
*/
|
||||
|
||||
|
||||
#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);
|
18
vs_win/include/reader.hpp
Normal file
18
vs_win/include/reader.hpp
Normal file
@ -0,0 +1,18 @@
|
||||
/*
|
||||
* Functions for reading and processing txt file with floats
|
||||
* Tomas Goldmann,2024
|
||||
*/
|
||||
|
||||
|
||||
#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);
|
13
vs_win/include/utils.hpp
Normal file
13
vs_win/include/utils.hpp
Normal file
@ -0,0 +1,13 @@
|
||||
/*
|
||||
* Non-maximum suppression
|
||||
* Tomas Goldmann,2024
|
||||
*/
|
||||
|
||||
|
||||
|
||||
#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