ONNX
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.

Files changed (1) hide show
  1. demo.py +1 -1
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__ > "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)')
 
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)')