utils.py 文件源码

def reshape_maps_zoom( maps, new_dim, interp_order=1 ):
    res = []
    for k in range(maps.shape[0]):
        f_map = maps[k,:,:]
        scale = tuple(np.array(new_dim, dtype=float))/np.array(f_map.shape)
        out = zoom( f_map, scale, order=interp_order )
        res.append(  out  )
    return np.array(res)

# Reshape feature maps