--- a +++ b/options/test_options.py @@ -0,0 +1,12 @@ +from .base_options import BaseOptions + + +class TestOptions(BaseOptions): + def initialize(self, parser): + BaseOptions.initialize(self, parser) + parser.add_argument('--results_dir', type=str, default='../results/', help='saves results here.') + parser.add_argument('--isTrain', type=bool, default=False, help='load for training') + parser.add_argument('--pretrained_models_netS', type=str, default='./pretrained_models/carson_Jan2021.h5', help=' ') + parser.add_argument('--pretrained_models_netME', type=str, default='./pretrained_models/carmenJan2021.h5', help=' ') + parser.add_argument('--pipeline', type=str, default='segmentation_motion_strain', help=' ') + return parser \ No newline at end of file