demo / options /test_options.py
ybbwcwaps
Add FakeVideoDetect
e8e478e
raw
history blame
958 Bytes
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