There are 2 typos in https://github.com/abdullahtarek/tennis_analysis/blob/main/training/tennis_court_keypoints_training.ipynb
kps = np.array(items['kps']).flatten() should be kps = np.array(item['kps']).flatten()
devic = torch.device("cuda" if torch.cuda.is_available() else "cpu") should be device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
There are 2 typos in https://github.com/abdullahtarek/tennis_analysis/blob/main/training/tennis_court_keypoints_training.ipynb
kps = np.array(items['kps']).flatten()should bekps = np.array(item['kps']).flatten()devic = torch.device("cuda" if torch.cuda.is_available() else "cpu")should bedevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")