haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: yolov5 get width (#566)

Browse Source
This commit is contained in:
makaveli 2021-05-25 18:17:16 +05:30 committed by GitHub
parent d274d90722
commit 668d89bbd5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
yolov5/yolov5.cpp
View File

@ -1,5 +1,6 @@
#include <iostream>
#include <chrono>
#include <cmath>
#include "cuda_utils.h"
#include "logging.h"
#include "common.hpp"
@ -23,10 +24,8 @@ static Logger gLogger;
static int get_width(int x, float gw, int divisor = 8) {
//return math.ceil(x / divisor) * divisor
if (int(x * gw) % divisor == 0) {
return int(x * gw);
}
return (int(x * gw / divisor) + 1) * divisor;
return int(ceil((x * gw) / divisor)) * divisor;
}
static int get_depth(int x, float gd) {