feat : 支持上下左右同时镜像
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped

This commit is contained in:
zcr
2026-01-08 13:47:44 +08:00
parent 12f5ca3ca3
commit f6963070fb

View File

@@ -237,12 +237,21 @@ def update_base_size_priority(layers):
def transpose_rotate(layer, image): def transpose_rotate(layer, image):
transpose = layer.get('transpose', False) # transpose[0]是左右 transpose[1]是上下
transpose = layer.get('transpose', [0, 0])
rotate = layer.get('rotate', 0) rotate = layer.get('rotate', 0)
paste_x, paste_y = layer['adaptive_position'][1], layer['adaptive_position'][0] paste_x, paste_y = layer['adaptive_position'][1], layer['adaptive_position'][0]
if transpose:
flip = 0 if transpose == "left_right" else 1 # transpose左右是1 上下是-1
image = image.transpose(flip) if transpose[0] == 1:
# 左右
image = image.transpose(0)
if transpose[1] == 1:
# 上下
image = image.transpose(1)
if rotate: if rotate:
image = image.rotate(rotate, expand=True) image = image.rotate(rotate, expand=True)
# 4. 计算粘贴位置以保持视觉中心一致 # 4. 计算粘贴位置以保持视觉中心一致