retinaface代码讲解_retinaface pytorch代码解读(一)

prior_box.py

import torch

from itertools import product as product

import numpy as np

from math import ceil

class PriorBox(object):

def __init__(self, cfg, image_size=None, phase='train'):

super(PriorBox, self).__init__()

self.min_sizes = cfg['min_sizes']

self.steps = cfg['steps']

self.clip = cfg['clip']

self.image_size = image_size

self.feature_maps = [[ceil(self.image_size[0]/step), ceil(self.image_size[1]/step)] for step in self.steps]

self.name = "s"

def forward(self):

anchors = []

for k, f in enumerate(self.feature_maps):

min_sizes = self.min_sizes[k]

#print(product(range(f[0]), range(f[1])))

for i, j in product(range(f[0]), range(f[1])):

for min_size in min_sizes:

s_kx = min_si