Skip to content
Snippets Groups Projects
Commit 5c1fa2d9 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Modified config file, subselecting channels

parent 11003be4
Branches
Tags
1 merge request!18MCCNN trainer
......@@ -41,7 +41,7 @@ protocols="grandtest-color*depth*infrared*thermal-{}".format(frames) # makeup is
exlude_attacks_list=["makeup"]
SELECTED_CHANNELS = [0] # selects only color, depth and infrared
SELECTED_CHANNELS = [0,1,2] # selects only color, depth and infrared
img_transform_train = transforms.Compose([ChannelSelect(selected_channels = SELECTED_CHANNELS),transforms.ToPILImage(),transforms.RandomHorizontalFlip(),transforms.ToTensor()])# Add p=0.5 later
......@@ -67,7 +67,7 @@ dataset = DataFolder(data_folder=data_folder_train,
#==============================================================================
# Load the architecture
NUM_CHANNELS = 1
NUM_CHANNELS = 3
network=MCCNN(num_channels = NUM_CHANNELS)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment