File size: 512 Bytes
89cf463
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
# -- 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