WeSign/sign-predictor
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases 3 Wiki Activity

Basics sign predictor

Browse Source
This commit is contained in:
RobbeDeWaele
2023-04-08 12:52:20 +02:00
parent 7793122eac
commit 3716067f84
5 changed files with 389 additions and 1408 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
visualizations/webcam_view.py
View File

@@ -26,8 +26,8 @@ frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
keypoints = []
spoter_model = SPOTER(num_classes=26, hidden_dim=len(LANDMARKS) * 2)
spoter_model.load_state_dict(torch.load('models/spoter_76.pth', map_location=torch.device('cpu')))
spoter_model = SPOTER(num_classes=19, hidden_dim=len(LANDMARKS) * 2)
spoter_model.load_state_dict(torch.load('checkpoints/spoter_80.pth', map_location=torch.device('cpu')))
# get values of the landmarks as a list of integers
values = []