commit
78fe5fe47d
@ -0,0 +1 @@
|
||||
*.o
|
||||
@ -0,0 +1,21 @@
|
||||
{
|
||||
"configurations": [
|
||||
{
|
||||
"name": "Win32",
|
||||
"includePath": [
|
||||
"${workspaceFolder}/**",
|
||||
"${workspaceFolder}/../include"
|
||||
],
|
||||
"defines": [
|
||||
"_DEBUG",
|
||||
"UNICODE",
|
||||
"_UNICODE"
|
||||
],
|
||||
"compilerPath": "g++",
|
||||
"cStandard": "c11",
|
||||
"cppStandard": "c++17",
|
||||
"intelliSenseMode": "gcc-x64"
|
||||
}
|
||||
],
|
||||
"version": 4
|
||||
}
|
||||
@ -0,0 +1,38 @@
|
||||
{
|
||||
"version": "0.2.0",
|
||||
"configurations": [
|
||||
{
|
||||
"name": "(gdb) Launch",
|
||||
"type": "cppdbg",
|
||||
"request": "launch",
|
||||
"program": "${workspaceFolder}/retinanetpost",
|
||||
"args": ["input/input.jpg", "input/input.txt"],
|
||||
"stopAtEntry": true,
|
||||
"cwd": "${workspaceFolder}",
|
||||
"environment": [],
|
||||
"externalConsole": false,
|
||||
"preLaunchTask": "Build IPA_projekt_2024",
|
||||
"windows": {
|
||||
"MIMode": "gdb",
|
||||
"miDebuggerPath": "gdb",
|
||||
"setupCommands": [
|
||||
{
|
||||
"description": "Enable pretty-printing for gdb",
|
||||
"text": "-enable-pretty-printing",
|
||||
"ignoreFailures": true
|
||||
},
|
||||
{
|
||||
"description": "Reduce gdb verbosity",
|
||||
"text": "set print thread-events on",
|
||||
"ignoreFailures": true
|
||||
}
|
||||
],
|
||||
"logging": {
|
||||
"trace": true,
|
||||
"traceResponse": true,
|
||||
"engineLogging": true
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
@ -0,0 +1,68 @@
|
||||
{
|
||||
"files.associations": {
|
||||
"iostream": "cpp",
|
||||
"array": "cpp",
|
||||
"atomic": "cpp",
|
||||
"bit": "cpp",
|
||||
"*.tcc": "cpp",
|
||||
"cctype": "cpp",
|
||||
"chrono": "cpp",
|
||||
"clocale": "cpp",
|
||||
"cmath": "cpp",
|
||||
"compare": "cpp",
|
||||
"complex": "cpp",
|
||||
"concepts": "cpp",
|
||||
"condition_variable": "cpp",
|
||||
"cstdarg": "cpp",
|
||||
"cstddef": "cpp",
|
||||
"cstdint": "cpp",
|
||||
"cstdio": "cpp",
|
||||
"cstdlib": "cpp",
|
||||
"cstring": "cpp",
|
||||
"ctime": "cpp",
|
||||
"cwchar": "cpp",
|
||||
"cwctype": "cpp",
|
||||
"deque": "cpp",
|
||||
"list": "cpp",
|
||||
"map": "cpp",
|
||||
"set": "cpp",
|
||||
"string": "cpp",
|
||||
"unordered_map": "cpp",
|
||||
"unordered_set": "cpp",
|
||||
"vector": "cpp",
|
||||
"exception": "cpp",
|
||||
"algorithm": "cpp",
|
||||
"functional": "cpp",
|
||||
"iterator": "cpp",
|
||||
"memory": "cpp",
|
||||
"memory_resource": "cpp",
|
||||
"numeric": "cpp",
|
||||
"random": "cpp",
|
||||
"ratio": "cpp",
|
||||
"string_view": "cpp",
|
||||
"system_error": "cpp",
|
||||
"tuple": "cpp",
|
||||
"type_traits": "cpp",
|
||||
"utility": "cpp",
|
||||
"fstream": "cpp",
|
||||
"initializer_list": "cpp",
|
||||
"iomanip": "cpp",
|
||||
"iosfwd": "cpp",
|
||||
"istream": "cpp",
|
||||
"limits": "cpp",
|
||||
"mutex": "cpp",
|
||||
"new": "cpp",
|
||||
"numbers": "cpp",
|
||||
"ostream": "cpp",
|
||||
"semaphore": "cpp",
|
||||
"sstream": "cpp",
|
||||
"stdexcept": "cpp",
|
||||
"stop_token": "cpp",
|
||||
"streambuf": "cpp",
|
||||
"thread": "cpp",
|
||||
"cinttypes": "cpp",
|
||||
"typeindex": "cpp",
|
||||
"typeinfo": "cpp"
|
||||
},
|
||||
"debug.allowBreakpointsEverywhere": true
|
||||
}
|
||||
@ -0,0 +1,78 @@
|
||||
{
|
||||
"version": "2.0.0",
|
||||
"windows": {
|
||||
|
||||
},
|
||||
"tasks": [
|
||||
{
|
||||
"label": "Build IPA_projekt_2024",
|
||||
"type": "shell",
|
||||
"command": "g++",
|
||||
"args": [
|
||||
"-g",
|
||||
"-std=c++11",
|
||||
"-fpic",
|
||||
"-o",
|
||||
"retinanetpost",
|
||||
"-DDEBUG",
|
||||
"retinanetpost_asm.o",
|
||||
"retinanetpost.cpp",
|
||||
"src/utils.cpp",
|
||||
"src/prior_boxes.cpp",
|
||||
"src/reader.cpp",
|
||||
"-Iinclude/",
|
||||
"`pkg-config", "--cflags", "opencv4`",
|
||||
"`pkg-config", "--libs", "opencv4`",
|
||||
"-ldl"
|
||||
],
|
||||
"group": {
|
||||
"kind": "build",
|
||||
"isDefault": true
|
||||
},
|
||||
"problemMatcher": [],
|
||||
"dependsOn": [
|
||||
"ASM part"
|
||||
],
|
||||
"isBackground": false
|
||||
},
|
||||
{
|
||||
"label": "ASM part",
|
||||
"type": "shell",
|
||||
"command": "gcc",
|
||||
"args": [
|
||||
"-masm=intel",
|
||||
"-g",
|
||||
"-c",
|
||||
"retinanetpost_asm.s"
|
||||
],
|
||||
"group": {
|
||||
"kind": "build",
|
||||
"isDefault": true
|
||||
},
|
||||
"problemMatcher": [],
|
||||
"dependsOn": [
|
||||
|
||||
],
|
||||
"isBackground": false
|
||||
},
|
||||
{
|
||||
"label": "Run",
|
||||
"type": "shell",
|
||||
"command": "./retinanetpost",
|
||||
"args": [
|
||||
"input/input.jpg",
|
||||
|
||||
"input/input.txt"
|
||||
],
|
||||
"group": {
|
||||
"kind": "build",
|
||||
"isDefault": true
|
||||
},
|
||||
"problemMatcher": [],
|
||||
"dependsOn": [
|
||||
"Build IPA_projekt_2024"
|
||||
],
|
||||
"isBackground": false
|
||||
}
|
||||
]
|
||||
}
|
||||
@ -0,0 +1,32 @@
|
||||
# Makefile for IPA_projekt_2024
|
||||
|
||||
CXX = g++
|
||||
CXXFLAGS = -g -std=c++11 -fpic `pkg-config --cflags opencv4`
|
||||
LDFLAGS = -ldl
|
||||
LIBS = `pkg-config --cflags --libs opencv4`
|
||||
INCLUDES = -Iinclude/
|
||||
|
||||
SRCS = retinanetpost.cpp src/utils.cpp src/prior_boxes.cpp src/reader.cpp
|
||||
ASMSRC = retinanetpost_asm.s
|
||||
OBJS = $(SRCS:.cpp=.o) $(ASMSRC:.s=.o)
|
||||
TARGET = retinanetpost
|
||||
#DEBUG = -DDEBUG
|
||||
|
||||
.PHONY: all clean
|
||||
|
||||
all: $(TARGET)
|
||||
|
||||
$(TARGET): $(OBJS)
|
||||
$(CXX) $(CXXFLAGS) $(DEBUG) -o $@ $^ $(LIBS) $(LDFLAGS)
|
||||
|
||||
%.o: %.cpp
|
||||
$(CXX) $(CXXFLAGS) $(DEBUG) $(INCLUDES) -c $< -o $@
|
||||
|
||||
%.o: %.s
|
||||
gcc -masm=intel -g -c $< -o $@
|
||||
|
||||
clean:
|
||||
$(RM) $(OBJS) $(TARGET)
|
||||
|
||||
run: $(TARGET)
|
||||
./$(TARGET) input/input.jpg input/input.txt
|
||||
@ -0,0 +1,15 @@
|
||||
#Not used for project
|
||||
|
||||
[BITS 64]
|
||||
|
||||
global f1:function
|
||||
|
||||
|
||||
section .data
|
||||
|
||||
|
||||
section .text
|
||||
|
||||
f1:
|
||||
mov rax, 0
|
||||
ret
|
||||
@ -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);
|
||||
@ -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);
|
||||
@ -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);
|
||||
|
After Width: | Height: | Size: 284 KiB |
File diff suppressed because one or more lines are too long
@ -0,0 +1,35 @@
|
||||
#include <stdio.h>
|
||||
#include <inttypes.h>
|
||||
#include <x86intrin.h>
|
||||
|
||||
|
||||
class InstructionCounter
|
||||
{
|
||||
private:
|
||||
unsigned long long counter;
|
||||
unsigned long long end_count;
|
||||
unsigned long long total_count;
|
||||
public:
|
||||
|
||||
void start()
|
||||
{
|
||||
counter= __rdtsc();
|
||||
};
|
||||
|
||||
void end()
|
||||
{
|
||||
end_count = __rdtsc();
|
||||
};
|
||||
|
||||
void print()
|
||||
{
|
||||
printf("%I64lld \n", getCyclesCount());
|
||||
}
|
||||
|
||||
unsigned long long getCyclesCount()
|
||||
{
|
||||
end();
|
||||
total_count = end_count - counter;
|
||||
return total_count;
|
||||
}
|
||||
};
|
||||
@ -0,0 +1,116 @@
|
||||
|
||||
#include <cstring>
|
||||
#include <string>
|
||||
#include <iostream>
|
||||
#include <inttypes.h>
|
||||
#include <dlfcn.h>
|
||||
#include "ipa_tool.h"
|
||||
|
||||
#include <opencv2/highgui.hpp>
|
||||
#include <opencv2/core.hpp>
|
||||
#include <opencv2/imgproc.hpp>
|
||||
|
||||
using namespace std;
|
||||
using namespace cv;
|
||||
|
||||
#include "prior_boxes.hpp"
|
||||
#include "utils.hpp"
|
||||
#include "reader.hpp"
|
||||
|
||||
extern "C" { void f1(int a);}
|
||||
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
if (argc != 3)
|
||||
{
|
||||
std::cout << "Run program by: ./retinapost input/vector.txt input/image.png";
|
||||
}
|
||||
|
||||
Mat image = imread(argv[1]);
|
||||
|
||||
if (image.empty())
|
||||
{
|
||||
cout << "Could not open or find the image" << endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
//All constants refer to the configuration used in prior_boxes.cpp and to the 640x480 resolution.
|
||||
std::vector<int> image_size = {640, 480};
|
||||
std::vector<float> variances = {0.1f, 0.2f};
|
||||
size_t total0_len = 12600*4;
|
||||
size_t total1_len = 12600*2;
|
||||
size_t num_anchors = total0_len / 4;
|
||||
|
||||
PriorBox priorBox(image_size, "projekt");
|
||||
|
||||
std::vector<std::vector<float>> priors = priorBox.forward();
|
||||
|
||||
InstructionCounter counter;
|
||||
counter.start();
|
||||
/*******************Part to optmize*********************/
|
||||
|
||||
vector<float> floats = readFloatsFromFile(argv[2]);
|
||||
vector<vector<float>> split_vectors = splitFloats(floats, 12600*4);
|
||||
|
||||
vector<float> floatarr = split_vectors[0];
|
||||
vector<float> floatarrscr = split_vectors[1];
|
||||
|
||||
|
||||
std::vector<std::vector<float>> loc;
|
||||
for (size_t i = 0; i < num_anchors; i++) {
|
||||
loc.push_back({floatarr[i * 4], floatarr[i * 4 + 1], floatarr[i * 4 + 2], floatarr[i * 4 + 3]});
|
||||
}
|
||||
|
||||
std::vector<std::vector<float>> decoded_boxes = decode(loc, priors, variances);
|
||||
|
||||
std::vector<float> scores;
|
||||
std::vector<int> inds;
|
||||
std::vector<float> det_scores;
|
||||
|
||||
std::vector<std::vector<float>> det_boxes;
|
||||
|
||||
for (size_t i = 0; i < total1_len/2; i++) {
|
||||
scores.push_back(floatarrscr[i*2+1]);
|
||||
if(floatarrscr[i*2+1] > 0.999)
|
||||
{
|
||||
inds.push_back(i);
|
||||
decoded_boxes[i].push_back(floatarrscr[i*2+1]);
|
||||
decoded_boxes[i][0]= decoded_boxes[i][0]* 640;
|
||||
decoded_boxes[i][1]= decoded_boxes[i][1]* 480;
|
||||
decoded_boxes[i][2]= decoded_boxes[i][2]* 640;
|
||||
decoded_boxes[i][3]= decoded_boxes[i][3]* 480;
|
||||
det_boxes.push_back(decoded_boxes[i]);
|
||||
det_scores.push_back(scores[i]);
|
||||
}
|
||||
}
|
||||
|
||||
auto out = nms(det_boxes, 0.4);
|
||||
|
||||
f1(10);
|
||||
|
||||
counter.print();
|
||||
|
||||
|
||||
/************************************************/
|
||||
|
||||
|
||||
Scalar color(0, 255, 0); // Color of the rectangle (in BGR)
|
||||
int thickness = 2; // Thickness of the rectangle border
|
||||
|
||||
for (int i = 0; i < out.size(); i++)
|
||||
{
|
||||
#ifdef DEBUG
|
||||
printf("Box %f %f %f %f %f\n", out[i][0], out[i][1], out[i][2], out[i][3], out[i][4]);
|
||||
#endif
|
||||
|
||||
|
||||
cv::Rect roi((int)out[i][0], (int)out[i][1], (int)out[i][2]- (int)out[i][0], (int)out[i][3] - (int)out[i][1]);
|
||||
rectangle(image, roi, color, thickness);
|
||||
}
|
||||
|
||||
imshow("Output", image);
|
||||
waitKey(0);
|
||||
|
||||
return 0;
|
||||
}
|
||||
@ -0,0 +1,13 @@
|
||||
.intel_syntax noprefix
|
||||
|
||||
.data
|
||||
constants: .byte 50,50,50,50
|
||||
|
||||
.text
|
||||
|
||||
.global f1
|
||||
f1:
|
||||
|
||||
ret 0
|
||||
|
||||
|
||||
@ -0,0 +1,109 @@
|
||||
/*
|
||||
* Prior boxes for RetinaNet
|
||||
* Tomas Goldmann,2023
|
||||
*/
|
||||
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <cmath>
|
||||
#include <map>
|
||||
#include <algorithm>
|
||||
#include "prior_boxes.hpp"
|
||||
|
||||
PriorBox::PriorBox(std::vector<int> image_size, std::string phase)
|
||||
{
|
||||
|
||||
std::map<std::string, std::vector<std::vector<float>>> cfg;
|
||||
this->min_sizes = {{16.0f, 32.0f}, {64.0f, 128.0f}, {256.0f, 512.0f}};
|
||||
this->image_size = {image_size[1], image_size[0]};
|
||||
this->steps = {8, 16, 32};
|
||||
|
||||
for (const float step : steps)
|
||||
{
|
||||
this->feature_maps.push_back({(int)std::ceil(image_size[0] / step), (int)std::ceil(image_size[1] / step)});
|
||||
}
|
||||
}
|
||||
|
||||
std::vector<std::vector<float>> PriorBox::forward()
|
||||
{
|
||||
|
||||
std::vector<float> anchors;
|
||||
for (size_t k = 0; k < this->feature_maps.size(); k++)
|
||||
{
|
||||
const auto &f = this->feature_maps[k];
|
||||
const auto &min_sizes = this->min_sizes[k];
|
||||
|
||||
for (int i = 0; i < f[1]; i++)
|
||||
{
|
||||
for (int j = 0; j < f[0]; j++)
|
||||
{
|
||||
for (const auto &min_size : min_sizes)
|
||||
{
|
||||
float s_kx = min_size / this->image_size[1];
|
||||
float s_ky = min_size / this->image_size[0];
|
||||
std::vector<float> dense_cx = {static_cast<float>(j + 0.5) * this->steps[k] / this->image_size[1]};
|
||||
std::vector<float> dense_cy = {static_cast<float>(i + 0.5) * this->steps[k] / this->image_size[0]};
|
||||
|
||||
for (const auto &cy : dense_cy)
|
||||
{
|
||||
for (const auto &cx : dense_cx)
|
||||
{
|
||||
|
||||
anchors.push_back(cx);
|
||||
anchors.push_back(cy);
|
||||
anchors.push_back(s_kx);
|
||||
anchors.push_back(s_ky);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
std::vector<std::vector<float>> output;
|
||||
|
||||
size_t num_anchors = anchors.size() / 4;
|
||||
for (size_t i = 0; i < num_anchors; i++)
|
||||
{
|
||||
for (size_t j = 0; j < 4; j++)
|
||||
{
|
||||
anchors[i * 4 + j] = std::min(std::max(anchors[i * 4 + j], 0.0f), 1.0f);
|
||||
}
|
||||
output.push_back({anchors[i * 4], anchors[i * 4 + 1], anchors[i * 4 + 2], anchors[i * 4 + 3]});
|
||||
}
|
||||
//std::cout << "len " << output.size() << std::endl;
|
||||
|
||||
return output;
|
||||
}
|
||||
|
||||
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)
|
||||
{
|
||||
std::vector<std::vector<float>> boxes;
|
||||
for (size_t i = 0; i < loc.size(); i++)
|
||||
{
|
||||
const auto &prior = priors[i];
|
||||
const auto &loc_pred = loc[i];
|
||||
float prior_x = prior[0];
|
||||
float prior_y = prior[1];
|
||||
float prior_w = prior[2];
|
||||
float prior_h = prior[3];
|
||||
float var_x = variances[0];
|
||||
float var_y = variances[1];
|
||||
float var_w = variances[2];
|
||||
float var_h = variances[3];
|
||||
|
||||
float decoded_x = prior_x + loc_pred[0] * var_x * prior_w;
|
||||
float decoded_y = prior_y + loc_pred[1] * var_y * prior_h;
|
||||
float decoded_w = prior_w * std::exp(loc_pred[2] * var_w);
|
||||
float decoded_h = prior_h * std::exp(loc_pred[3] * var_h);
|
||||
|
||||
float decoded_xmin = decoded_x - decoded_w / 2;
|
||||
float decoded_ymin = decoded_y - decoded_h / 2;
|
||||
float decoded_xmax = decoded_x + decoded_w / 2;
|
||||
float decoded_ymax = decoded_y + decoded_h / 2;
|
||||
|
||||
boxes.push_back({decoded_xmin, decoded_ymin, decoded_xmax, decoded_ymax});
|
||||
}
|
||||
|
||||
return boxes;
|
||||
}
|
||||
@ -0,0 +1,61 @@
|
||||
/*
|
||||
* Function to read floats from txt
|
||||
* Tomas Goldmann,2023
|
||||
*/
|
||||
|
||||
|
||||
#include "reader.hpp"
|
||||
|
||||
using namespace std;
|
||||
|
||||
vector<vector<float>> splitFloats(const vector<float>& floats, int index) {
|
||||
|
||||
vector<float> first_half;
|
||||
vector<float> second_half;
|
||||
|
||||
// Ensure the original vector has enough elements for splitting
|
||||
if (floats.size() < 6*12600) {
|
||||
cerr << "Error: The input must containt 6*12600 floats" << endl;
|
||||
return {first_half, second_half}; // Return empty vectors
|
||||
}
|
||||
|
||||
// Split the original vector into two separate vectors
|
||||
first_half.insert(first_half.end(), floats.begin(), floats.begin() + index);
|
||||
second_half.insert(second_half.end(), floats.begin() + index, floats.end() );
|
||||
|
||||
// Return a vector containing the two split vectors
|
||||
return {first_half, second_half};
|
||||
}
|
||||
|
||||
|
||||
|
||||
vector<float> readFloatsFromFile(const string& filename) {
|
||||
vector<float> floats;
|
||||
ifstream file(filename);
|
||||
|
||||
// Check if the file is opened successfully
|
||||
if (!file.is_open()) {
|
||||
cerr << "Error opening the file" << endl;
|
||||
return floats; // Return an empty vector if the file cannot be opened
|
||||
}
|
||||
|
||||
string line;
|
||||
|
||||
// Read each line from the file
|
||||
while (getline(file, line)) {
|
||||
// Create a string stream from the line
|
||||
stringstream ss(line);
|
||||
string token;
|
||||
|
||||
// Split the line by commas and read each float
|
||||
while (getline(ss, token, ',')) {
|
||||
// Convert the string to a float and store it in the vector
|
||||
floats.push_back(stof(token));
|
||||
}
|
||||
}
|
||||
|
||||
// Close the file
|
||||
file.close();
|
||||
|
||||
return floats;
|
||||
}
|
||||
@ -0,0 +1,64 @@
|
||||
#include "utils.hpp"
|
||||
|
||||
|
||||
bool CompareBBox(const std::vector<float> & a, const std::vector<float> & b)
|
||||
{
|
||||
return a[4] > b[4];
|
||||
}
|
||||
|
||||
|
||||
|
||||
std::vector<std::vector<float>> nms(std::vector<std::vector<float>>& bboxes, float threshold)
|
||||
{
|
||||
std::vector<std::vector<float>> bboxes_nms;
|
||||
std::sort(bboxes.begin(), bboxes.end(), CompareBBox);
|
||||
|
||||
int32_t select_idx = 0;
|
||||
int32_t num_bbox = static_cast<int32_t>(bboxes.size());
|
||||
std::vector<int32_t> mask_merged(num_bbox, 0);
|
||||
bool all_merged = false;
|
||||
|
||||
while (!all_merged) {
|
||||
while (select_idx < num_bbox && mask_merged[select_idx] == 1)
|
||||
select_idx++;
|
||||
if (select_idx == num_bbox) {
|
||||
all_merged = true;
|
||||
continue;
|
||||
}
|
||||
|
||||
bboxes_nms.push_back(bboxes[select_idx]);
|
||||
mask_merged[select_idx] = 1;
|
||||
|
||||
std::vector<float> select_bbox = bboxes[select_idx];
|
||||
float area1 = static_cast<float>((select_bbox[2] - select_bbox[0] + 1) * (select_bbox[3] - select_bbox[1] + 1));
|
||||
float x1 = static_cast<float>(select_bbox[0]);
|
||||
float y1 = static_cast<float>(select_bbox[1]);
|
||||
float x2 = static_cast<float>(select_bbox[2]);
|
||||
float y2 = static_cast<float>(select_bbox[3]);
|
||||
|
||||
select_idx++;
|
||||
for (int32_t i = select_idx; i < num_bbox; i++) {
|
||||
if (mask_merged[i] == 1)
|
||||
continue;
|
||||
|
||||
std::vector<float>& bbox_i = bboxes[i];
|
||||
float x = std::max<float>(x1, static_cast<float>(bbox_i[0]));
|
||||
float y = std::max<float>(y1, static_cast<float>(bbox_i[1]));
|
||||
float w = std::min<float>(x2, static_cast<float>(bbox_i[2])) - x + 1;
|
||||
float h = std::min<float>(y2, static_cast<float>(bbox_i[3])) - y + 1;
|
||||
|
||||
if (w <= 0 || h <= 0)
|
||||
continue;
|
||||
|
||||
float area2 = static_cast<float>((bbox_i[2] - bbox_i[0] + 1) * (bbox_i[3] - bbox_i[1] + 1));
|
||||
float area_intersect = w * h;
|
||||
|
||||
|
||||
if (static_cast<float>(area_intersect) / (area1 + area2 - area_intersect) > threshold) {
|
||||
mask_merged[i] = 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return bboxes_nms;
|
||||
}
|
||||
Loading…
Reference in new issue