Skip to content
Snippets Groups Projects
Commit 2a2d9b20 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[script] add world only option on training script for DR-GAN on Multi-PIE

parent c7be844f
No related branches found
No related tags found
No related merge requests found
......@@ -122,7 +122,8 @@ def main(user_input=None):
# WARNING with the transforms ... act on labels too, at some point, I may have to write my own
# Also, in 'ToTensor', there is a reshape performed from: HxWxC to CxHxW
face_dataset = MultiPIEDataset(root_dir=multipie_root_dir,
frontal_only=False,
frontal_only=False,
world=False,
transform=transforms.Compose([
RollChannels(), # bob to skimage:
ToTensor(),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment