From ec776eac6feaf59b4860678c3363ce7634e19c5e Mon Sep 17 00:00:00 2001
From: Rex <hkchengrex@gmail.com>
Date: Wed, 6 Jul 2022 17:58:32 -0500
Subject: [PATCH] fixes

---
 inference/interact/gui.py    | 4 ++++
 interactive_demo.py          | 3 ---
 requirements_demo.txt        | 3 ++-
 scripts/download_datasets.py | 3 ++-
 4 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/inference/interact/gui.py b/inference/interact/gui.py
index 242888c8..8703ea92 100644
--- a/inference/interact/gui.py
+++ b/inference/interact/gui.py
@@ -14,7 +14,11 @@
 
 import functools
 
+import os
 import cv2
+# fix conflicts between qt5 and cv2
+os.environ.pop("QT_QPA_PLATFORM_PLUGIN_PATH")
+
 import numpy as np
 import torch
 
diff --git a/interactive_demo.py b/interactive_demo.py
index 6e899eca..92953f31 100644
--- a/interactive_demo.py
+++ b/interactive_demo.py
@@ -3,13 +3,10 @@
 """
 
 import sys
-import os
 from os import path
-import functools
 from argparse import ArgumentParser
 
 import torch
-from yaml import parse
 
 from model.network import XMem
 from inference.interact.s2m_controller import S2MController
diff --git a/requirements_demo.txt b/requirements_demo.txt
index 994c34ea..ad3d04cb 100644
--- a/requirements_demo.txt
+++ b/requirements_demo.txt
@@ -1,2 +1,3 @@
 PyQt5
-Cython
\ No newline at end of file
+Cython
+scipy
\ No newline at end of file
diff --git a/scripts/download_datasets.py b/scripts/download_datasets.py
index bd8e7cc1..7537aead 100644
--- a/scripts/download_datasets.py
+++ b/scripts/download_datasets.py
@@ -21,10 +21,11 @@
 YouTubeVOS: https://youtube-vos.org
 DAVIS: https://davischallenge.org/
 BL30K: https://github.com/hkchengrex/MiVOS
+Long-Time Video: https://github.com/xmlyqing00/AFB-URR
 """
 
 print(LICENSE)
-print('Datasets will be downloaded and extracted to ../YouTube, ../YouTube2018, ../static, ../DAVIS')
+print('Datasets will be downloaded and extracted to ../YouTube, ../YouTube2018, ../static, ../DAVIS, ../long_video_set')
 reply = input('[y] to confirm, others to exit: ')
 if reply != 'y':
     exit()