def __init__(self):
TestOptions.__init__(self)
self.parser.add_argument('--root-dir',
help='dir containing the two classes folders',
dest="root_dir")
self.parser.add_argument('--gpu',
type=bool,
default=False,
help='Train on GPU')
def __init__(self):
TestOptions.__init__(self)
self.parser.add_argument('--gpu',
type=bool,
default=False,
help='Train on GPU')
self.parser.add_argument('--flat',
type=bool,
default=True,
help='Flat folder structure')
