Skip to content

SSD 格式转换



格式转换¶

格式转换包括了3个实现：

ConvertFromInts
ToPercentCoords
ToTensor

ConvertFromInts¶

py/ssd/data/transforms/convert_from_ints.py

将数据精度由uint8转换为float32

ToPercentCoords¶

py/ssd/data/transforms/to_percent_coords.py

该实现是针对边界框坐标实现的。其目的是将边界框绝对值坐标转换为相对于图像宽高的相对值，这样以便于后续的图像缩放和边界框检测器训练

ToTensor¶

py/ssd/data/transforms/to_tensor.py

将Numpy数组转换成PyTorch张量，并调整维度排列：[H, W, C] -> [C, H, W]