YOLOv3学习笔记之数据处理
dataset中的数据处理
1. 图片处理
将读取的图片进行reshape,变为正方形
def pad_to_square(img, pad_value):
c, h, w = img.shape
dim_diff = np.abs(h - w)
# (upper / left) padding and (lower / right) padding
pad1, pad2 = dim_diff // 2, dim_diff - dim_diff // 2
# Determine padding
pad = (0, 0, pad1, pad2) if h <= w else (pad1, pad2, 0, 0)
#(左填充,右填充,上填充,下填充)
# Add padding
img = F.pad(img, pad, "constant"