# -- coding: utf-8 -- # @Time : 2022/8/26 # @Author : ykk648 # @Project : https://github.com/ykk648/AI_power # def reverse2wholeimage_hifi(swaped_img, mat_rev, img_mask, frame_wait_merge, orisize): # swaped_img = swaped_img.cpu().numpy().transpose((1, 2, 0)) # target_image = cv2.warpAffine(swaped_img, mat_rev, orisize) # img = ne.evaluate('img_mask * (target_image * 255) ')[..., ::-1] # img = ne.evaluate('img + frame_wait_merge') # final_img = img.astype(np.uint8) # return final_img