def __init__(self):
# datasets
self.train_dataset = Dataset.UCF101VideoDataset(
'/home/sunhanbo/workspace/C3D_UCF101/Dataset', 'train', 16, False)
self.train_loader = Data.DataLoader(
self.train_dataset,
batch_size=60,
shuffle=True,
drop_last=True,
num_workers=16,
)
self.test_dataset = Dataset.UCF101VideoDataset(
'/home/sunhanbo/workspace/C3D_UCF101/Dataset', 'test', 16, False)
self.test_loader = Data.DataLoader(
self.test_dataset,
batch_size=60,
shuffle=False,
drop_last=False,
num_workers=16,
)
# networks
self.net = Network.C3DNetwork(101)
