cv2

app.py

@@ -31,7 +31,6 @@ from misc.utils import show_imgs
 import sys 
 from misc.dataset import TextEncoder 
 import requests 
-import cv2 
 from io import BytesIO
 from translate import Translator
 
@@ -54,8 +53,6 @@ def download_url_img(url):
         return False, []
     if response is not None and response.status_code == 200:
         input_image_data = response.content
-        # np_arr = np.asarray(bytearray(input_image_data), np.uint8).reshape(1, -1)
-        # parsed_image = cv2.imdecode(np_arr, cv2.IMREAD_UNCHANGED)
         image=Image.open(BytesIO(input_image_data))
         return True, image
     return False, []
@@ -74,7 +71,7 @@ def search(mode, text):
         with torch.no_grad():
             _, output_emb = join_emb(None, input_caps, length)
         caps_enc.append(output_emb.cpu().data.numpy())
-        
+
     caps_stack = np.vstack(caps_enc)
 
     imgs_url = [os.path.join("http://images.cocodataset.org/train2017", img_path.strip().split('_')[-1]) for img_path in imgs_path]

requirements.txt

@@ -9,5 +9,4 @@ torch==1.7.1
 torchvision==0.8.2
 tqdm==4.63.0
 translate==3.6.1
-visual_genome==1.1.1
-opencv-python
+visual_genome==1.1.1