Ryan Lee
commited on
Commit
·
24de7d3
1
Parent(s):
c5c8cff
Fix OpenCV version assertions in ViTTrack & RAFT and text offset (#234)
Browse files* Update OpenCV version assertions for Object Tracking and Optical Flow
* Increased text offset to not overlap with the bounding box.
demo.py
CHANGED
@@ -6,7 +6,7 @@ import numpy as np
|
|
6 |
from raft import Raft
|
7 |
|
8 |
# Check OpenCV version
|
9 |
-
assert cv.__version__
|
10 |
"Please install latest opencv-python to try this demo: python3 -m pip install --upgrade opencv-python"
|
11 |
|
12 |
parser = argparse.ArgumentParser(description='RAFT (https://github.com/princeton-vl/RAFT)')
|
|
|
6 |
from raft import Raft
|
7 |
|
8 |
# Check OpenCV version
|
9 |
+
assert cv.__version__ >= "4.9.0", \
|
10 |
"Please install latest opencv-python to try this demo: python3 -m pip install --upgrade opencv-python"
|
11 |
|
12 |
parser = argparse.ArgumentParser(description='RAFT (https://github.com/princeton-vl/RAFT)')
|