def __init__(self, image_dir):
super(Dataset, self).__init__()
# self.path = image_dir
self.input_filenames = glob.glob(os.path.join(image_dir, "*.jpg"))
微信
公众号
扫码关注公众号