File size: 708 Bytes
5db43ff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import argparse
import os
class BaseOptions():
def __init__(self):
self.parser = argparse.ArgumentParser()
self.initialized = False
def initialize(self):
self.parser.add_argument('--video_path', type=str, default='./example_video.mp4',help='the path of video')
self.parser.add_argument('--mask_dir', type=str, default='./example_video',help='the path of mask images')
self.parser.add_argument('--dataset_name', type=str, default='example_dataset', help='name of the dataset')
self.initialized=True
def parse(self):
if not self.initialized:
self.initialize()
self.opt = self.parser.parse_args()
return self.opt |