|
|
import cv2
|
|
|
import numpy as np
|
|
|
|
|
|
from img_1_color import *
|
|
|
from img_2_operation import *
|
|
|
from img_3_transformation import *
|
|
|
from img_4_noise import *
|
|
|
from img_5_enhance import *
|
|
|
from img_6_morphology import *
|
|
|
from img_styleTransfer import *
|
|
|
from img_aging import *
|
|
|
|
|
|
|
|
|
def task(cmd, cnt):
|
|
|
# print(cmd)
|
|
|
if (cmd == '10'): # hsv_h
|
|
|
color_hsv(0, cnt)
|
|
|
elif (cmd == '11'): # hsv_s
|
|
|
color_hsv(1, cnt)
|
|
|
elif (cmd == '12'): # hsv_s
|
|
|
color_hsv(2, cnt)
|
|
|
elif (cmd == '13'): # hsv_s
|
|
|
color_rgb(0, cnt)
|
|
|
elif (cmd == '14'): # hsv_s
|
|
|
color_rgb(1, cnt)
|
|
|
elif (cmd == '15'): # hsv_s
|
|
|
color_rgb(2, cnt)
|
|
|
|
|
|
elif (cmd == '20'): # 非运算
|
|
|
operate(0, cnt)
|
|
|
elif (cmd == '21'): # 或运算
|
|
|
operate(1, cnt)
|
|
|
elif (cmd == '22'): # 与运算
|
|
|
operate(2, cnt)
|
|
|
elif (cmd == '23'): # 加运算
|
|
|
operate(3, cnt)
|
|
|
elif (cmd == '24'): # 减运算
|
|
|
operate(4, cnt)
|
|
|
elif (cmd == '25'): # 乘运算
|
|
|
operate(5, cnt)
|
|
|
elif (cmd == '26'): # 除运算
|
|
|
operate(6, cnt)
|
|
|
|
|
|
elif (cmd == '30'): # 放大
|
|
|
transfer(0, cnt)
|
|
|
elif (cmd == '31'): # 缩小
|
|
|
transfer(1, cnt)
|
|
|
elif (cmd == '32'): # 旋转90°
|
|
|
transfer(2, cnt)
|
|
|
elif (cmd == '33'): # 水平镜像
|
|
|
transfer(3, cnt)
|
|
|
elif (cmd == '34'): # 垂直镜像
|
|
|
transfer(4, cnt)
|
|
|
|
|
|
elif (cmd == '40'): # 添加噪声
|
|
|
noise_add(cnt)
|
|
|
elif (cmd == '41'): # 减噪
|
|
|
noise_filter(cnt)
|
|
|
|
|
|
elif (cmd == '50'): # 白平衡
|
|
|
enhance(0, cnt)
|
|
|
elif (cmd == '51'): # 灰度世界算法
|
|
|
enhance(1, cnt)
|
|
|
elif (cmd == '52'): # 直方图均衡化
|
|
|
enhance(2, cnt)
|
|
|
elif (cmd == '53'): # Retinex ssr1
|
|
|
enhance(3, cnt)
|
|
|
elif (cmd == '54'): # Retinex ssr2
|
|
|
enhance(4, cnt)
|
|
|
elif (cmd == '55'): # Retinex msr1
|
|
|
enhance(5, cnt)
|
|
|
elif (cmd == '56'): # Retinex msr2
|
|
|
enhance(6, cnt)
|
|
|
elif (cmd == '57'): # Retinex msrcr
|
|
|
enhance(7, cnt)
|
|
|
elif (cmd == '58'): # 自动白平衡（注：用时较长）
|
|
|
enhance(8, cnt)
|
|
|
elif (cmd == '59'): # 自动色彩均衡（注：用时较长）
|
|
|
enhance(9, cnt)
|
|
|
|
|
|
elif (cmd == '60'): # 腐蚀
|
|
|
morphology(0, cnt)
|
|
|
elif (cmd == '61'): # 膨胀
|
|
|
morphology(1, cnt)
|
|
|
elif (cmd == '62'): # 开运算
|
|
|
morphology(2, cnt)
|
|
|
elif (cmd == '63'): # 闭运算
|
|
|
morphology(3, cnt)
|
|
|
|
|
|
# 风格迁移（注：需等待较长时间）
|
|
|
elif (cmd == '70'): # 风格：自定义
|
|
|
style_transfer(0, cnt)
|
|
|
elif (cmd == '71'): # 预设风格1：starsky
|
|
|
style_transfer(1, cnt)
|
|
|
elif (cmd == '72'): # 预设风格2：
|
|
|
style_transfer(2, cnt)
|
|
|
elif (cmd == '73'): # 预设风格3：
|
|
|
style_transfer(3, cnt)
|
|
|
|
|
|
# 年龄转换
|
|
|
elif (cmd == '80'):
|
|
|
Aging(0, cnt) #此时启用男性年龄模型
|
|
|
elif (cmd == '81'):
|
|
|
Aging(1, cnt) #此时启用女性年龄模型
|
|
|
|
|
|
else:
|
|
|
print('Invalid command!')
|
|
|
return 1
|
|
|
|
|
|
return 0
|
