Skip to content

Commit

Permalink
fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
hkchengrex committed Jul 6, 2022
1 parent 6b7491a commit ec776ea
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 5 deletions.
4 changes: 4 additions & 0 deletions inference/interact/gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
3 changes: 0 additions & 3 deletions interactive_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
3 changes: 2 additions & 1 deletion requirements_demo.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
PyQt5
Cython
Cython
scipy
3 changes: 2 additions & 1 deletion scripts/download_datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down

0 comments on commit ec776ea

Please sign in to comment.