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):
|
||||
transpose = layer.get('transpose', False)
|
||||
# transpose[0]是左右 transpose[1]是上下
|
||||
transpose = layer.get('transpose', [0, 0])
|
||||
|
||||
rotate = layer.get('rotate', 0)
|
||||
paste_x, paste_y = layer['adaptive_position'][1], layer['adaptive_position'][0]
|
||||
if transpose:
|
||||
flip = 0 if transpose == "left_right" else 1
|
||||
image = image.transpose(flip)
|
||||
|
||||
# transpose左右是1 上下是-1
|
||||
if transpose[0] == 1:
|
||||
# 左右
|
||||
image = image.transpose(0)
|
||||
|
||||
if transpose[1] == 1:
|
||||
# 上下
|
||||
image = image.transpose(1)
|
||||
|
||||
if rotate:
|
||||
image = image.rotate(rotate, expand=True)
|
||||
# 4. 计算粘贴位置以保持视觉中心一致
|
||||
|
||||
Reference in New Issue
Block a user