face_enhancement.py

@@ -12,7 +12,6 @@ import __init_paths
 from retinaface.retinaface_detection import RetinaFaceDetection
 from face_model.face_gan import FaceGAN
 from align_faces import warp_and_crop_face, get_reference_facial_points
-from google.colab.patches import cv2_imshow
 
 class FaceEnhancement(object):
     def __init__(self, base_dir='./', size=512, out_size=None, model=None, channel_multiplier=2, narrow=1, key=None, device='cuda'):