search

abdullahtarek / tennis_analysis

This project analyzes Tennis players in a video to measure their speed, ball shot speed and number of shots. This project will detect players and the tennis ball using YOLO and also utilizes CNNs to extract court keypoints. This hands on project is perfect for polishing your machine learning, and computer vision skills.
405 stars 140 forks source link

Typos in tennis_court_keypoints_training.ipynb #5

Open lucarubini opened 5 months ago

lucarubini commented 5 months ago

There are 2 typos in https://github.com/abdullahtarek/tennis_analysis/blob/main/training/tennis_court_keypoints_training.ipynb

  1. kps = np.array(items['kps']).flatten() should be kps = np.array(item['kps']).flatten()
  2. devic = torch.device("cuda" if torch.cuda.is_available() else "cpu")should be device = torch.device("cuda" if torch.cuda.is_available() else "cpu")