未验证 提交 e4226f57 编写于 作者: A AddSalt8227 提交者: GitHub

fix yolov4-tiny anchors bug (#1307)

* std::max_element in range [first, last)

* Comment out misleading output

* fix yolov4-tiny anchors bug
上级 4fb2e3c3
......@@ -168,7 +168,7 @@ void get_input_data_yolov4(const char* image_file, float* input_data, int img_h,
static void generate_proposals(int stride, const float* feat, float prob_threshold, std::vector<Object>& objects)
{
static float anchors[12] = {10, 14, 23, 27, 37, 58, 81, 82, 135, 169, 344, 319};
static float anchors[12] = {23, 27, 37, 58, 81, 82, 81, 82, 135, 169, 344, 319};
int anchor_num = 3;
int feat_w = 416 / stride;
int feat_h = 416 / stride;
......
......@@ -176,7 +176,7 @@ void get_input_data_yolov4_uint8(const char* image_file, uint8_t* input_data, in
static void generate_proposals(int stride, const float* feat, float prob_threshold, std::vector<Object>& objects)
{
static float anchors[12] = {10, 14, 23, 27, 37, 58, 81, 82, 135, 169, 344, 319};
static float anchors[12] = {23, 27, 37, 58, 81, 82, 81, 82, 135, 169, 344, 319};
int anchor_num = 3;
int feat_w = 416 / stride;
int feat_h = 416 / stride;
......
......@@ -176,7 +176,7 @@ void get_input_data_yolov4_uint8(const char* image_file, uint8_t* input_data, in
static void generate_proposals(int stride, const float* feat, float prob_threshold, std::vector<Object>& objects)
{
static float anchors[12] = {10, 14, 23, 27, 37, 58, 81, 82, 135, 169, 344, 319};
static float anchors[12] = {23, 27, 37, 58, 81, 82, 81, 82, 135, 169, 344, 319};
int anchor_num = 3;
int feat_w = 416 / stride;
int feat_h = 416 / stride;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册