yolo v3-------yolo_net 代码详解（一）

最新推荐文章于 2024-09-13 21:56:25 发布

幸运i

最新推荐文章于 2024-09-13 21:56:25 发布

阅读量1.5k

点赞数

本文链接：https://blog.csdn.net/weixin_40722688/article/details/88817879

版权

本文将逐步解析YOLOv3的实现代码yolo_net，深入理解其网络结构与关键操作。第一部分主要探讨网络架构及前向传播过程。

摘要由CSDN通过智能技术生成

持续更新中

第一部分：

import numpy as np
import tensorflow as tf
import yolo.config as cfg

slim = tf.contrib.slim

#创建YOLONet 类

class YOLONet(object):
    
    def __init__(self, is_training=True):
    '''
    构造函数
    利用cfg配置文件对网络参数进行初始化，定义网络的输入和输出等信息，
    '''
        self.classes = cfg.CLASSES
        self.num_class = len(self.classes)
        self.image_size = cfg.IMAGE_SIZE
        self.cell_size = cfg.CELL_SIZE    #将图片分为       
        self.boxes_per_cell = cfg.BOXES_PER_CELL #每个格子输出2个Bounding Box信息         
        self.output_size = (self.cell_size * self.cell_size) *\
            (self.num_class + self.boxes_per_cell * 5)
        self.scale = 1.0 * self.image_size / self.cell_size
        self.boundary1 = self.cell_size * self.cell_size * self.num_class
        self.boundary2 = self.boundary1 +\
            self.cell_size * self