Added README.md

Basket-Analytics · Apr 10, 2021 · abda958 · abda958
1 parent 1cdcb06
commit abda958
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -0,0 +1,5 @@
+# BasketTracking
+
+<p align='center'>
+ <img src="demo.gif" />
+</p>
diff --git a/demo.gif b/demo.gif
diff --git a/pano.png b/pano.png
diff --git a/pano_enhanced.png b/pano_enhanced.png
diff --git a/video_handler.py b/video_handler.py
@@ -10,6 +10,7 @@
 search_params = dict(checks=50)
 flann = cv2.FlannBasedMatcher(index_params, search_params)
 
+
 class VideoHandler:
  def __init__(self, pano, video, ball_detector, feet_detector, map_2d):
  self.M1 = np.load("Rectify1.npy")
@@ -29,14 +30,19 @@ def run_detectors(self):
  if not ok:
  break
  else:
- if 0 <= time_index < 20:
+ if 0 <= time_index < 200:
+
+ print("\r Computing DEMO: " + str(int(100 * time_index / 200)) + "%",
+ flush=True, end='')
+
  frame = frame[TOPCUT:, :]
  M = self.get_homography(frame, self.des1, self.kp1)
- frame, self.map_2d, map_2d_text = self.feet_detector.get_players_pos(M, self.M1, frame, time_index, self.map_2d)
+ frame, self.map_2d, map_2d_text = self.feet_detector.get_players_pos(M, self.M1, frame, time_index,
+ self.map_2d)
  frame, ball_map_2d = self.ball_detector.ball_tracker(M, self.M1, frame, self.map_2d.copy())
  vis = np.vstack((frame, cv2.resize(map_2d_text, (frame.shape[1], frame.shape[1] // 2))))
 
- cv2.imshow("Tracking", vis)
+ # cv2.imshow("Tracking", vis)
  writer.writeFrame(cv2.cvtColor(vis, cv2.COLOR_BGR2RGB))
 
  k = cv2.waitKey(1) & 0xff