
yolo
Rock_Huang~
AI Engineer
展开
-
yolov2-----源码阅读【reorg层】
官方yolov2 结构如图 reorg层的作用是将上一层卷积后的特征图38X38X64 拉伸成19X19X256,具体看代码: elif module_type == "reorg": stride = int(module["stride"]) #stride 为2 B,C,H,W = x.size() #B为bach大小暂设为32,c为通道数64 H,W都为38 x = ...原创 2019-07-24 19:20:38 · 1439 阅读 · 1 评论 -
YOLO系列的IOU、GIOU源码阅读
#include "box.h" #include <stdio.h> #include <math.h> #include <stdlib.h> //#define DEBUG_NAN //#define DEBUG_PRINTS int nms_comparator(const void *pa, const void *pb) { detect...原创 2019-09-25 10:22:06 · 2362 阅读 · 15 评论