feat : 支持上下左右同时镜像
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped
This commit is contained in:
@@ -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. 计算粘贴位置以保持视觉中心一致
|
||||||
|
|||||||
Reference in New Issue
Block a user