Spaces:
Sleeping
Sleeping
File size: 958 Bytes
3cc4a06 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from .base_options import BaseOptions
class TestOptions(BaseOptions):
def initialize(self, parser):
parser = BaseOptions.initialize(self, parser)
parser.add_argument('--model_path')
parser.add_argument('--no_resize', action='store_true')
parser.add_argument('--no_crop', action='store_true')
parser.add_argument('--eval', action='store_true', help='use eval mode during test time.')
parser.add_argument('--real_list_path', default='/mnt/data2/group2024-lhj/t2v/data/test/true', help='path for the list of real video,')
parser.add_argument('--fake_list_path', default='/mnt/data2/group2024-lhj/t2v/data/test/fake', help='path for the list of fake video,')
parser.add_argument('--ckpt', type=str, default='./checkpoints/best_network.pth')
parser.add_argument('--output', type=str, default='./checkpoints/test')
self.isTrain = False
return parser
|