测试数据集

需求：用已经训练好的模型测试各种数据集
有的有test.txt文件，可以从文件中获取图片地址，有的没有．每次都要改代码，头疼．写了一个迭代器

import glob

class imageInterator:
    #使用方法
    #i=imageInterator(path)
    #while i.hasNext()
    #    image_path=i.next()
    
    def __init__(self,path,root=""):
        self.image_path_list=[]
        if os.path.isdir(path):
            path_list=glob.glob(path+"/*.jpg")
            for image_path in path_list:
                self.image_path_list.append(image_path)

        if os.path.isfile(path):
            with open(path) as txtData:
                lines=txtData.readlines()
                for line in lines:
                    line=line.strip("\n")
                    if line.endswith(".jpg"):
                        image_path=os.path.join(root,line)
                    else:
                        image_path=os.path.join(root,line+".jpg")
                        
                    self.image_path_list.append(image_path)
        

    def next(self):
        return self.image_path_list.pop()

    def hasNext(self):
        if self.image_path_list:
            return True
        else:
            return False
    def get_list(self):
        return self.image_path_list