Implement pytorch dataset for own collected data

This commit is contained in:
Victor Mylle
2023-02-27 13:34:26 +00:00
parent 97ede38e3a
commit 8e5957f4ff
7 changed files with 99 additions and 7 deletions

View File

@@ -13,7 +13,7 @@ import torch.optim as optim
from torch.utils.data import DataLoader
from torchvision import transforms
from dataset import WLASLDataset
from datasets.wlasl_dataset import WLASLDataset
from identifiers import LANDMARKS
from keypoint_extractor import KeypointExtractor
from model import SPOTER