a b/options/test_options.py
1
from .base_options import BaseOptions
2
3
4
class TestOptions(BaseOptions):
5
    def initialize(self, parser):
6
        BaseOptions.initialize(self, parser)
7
        parser.add_argument('--results_dir', type=str, default='../results/', help='saves results here.')
8
        parser.add_argument('--isTrain', type=bool, default=False, help='load for training')
9
        parser.add_argument('--pretrained_models_netS', type=str, default='./pretrained_models/carson_Jan2021.h5', help=' ')
10
        parser.add_argument('--pretrained_models_netME', type=str, default='./pretrained_models/carmenJan2021.h5', help=' ')
11
        parser.add_argument('--pipeline', type=str, default='segmentation_motion_strain', help=' ')  
12
        return parser