ocr_datasets.py

import os
import glob
class OCRDatasets(object):
    def __init__(self):
        return
    def get_samples(self, split_name):
        sample_pattern = './imgs/{}/*.png'.format(split_name)
#         sample_pattern = './imgs/image_contest_level_1/*.png'.format(split_name)
        samples = glob.glob(sample_pattern)
       
        return samples
    def run(self):
        samples = self.get_samples('val')
        for sample in samples:
            print(sample)
        print(len(samples))
        return

if __name__ == "__main__":   
    obj= OCRDatasets()
    obj.run()