|
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 |
|
|