python iter和next使用 简单记录

文件格式：

# image/rgb/20210805/src/1628156293c9PUO9.jpg 1280 720
529 316 627 471 0.12109375 0
# image/rgb/20210805/src/1628156299YYfYJQ.jpg 1280 720
663 333 802 516 0.14296875 2
869 352 909 410 0.0453125 1
248 359 323 444 0.06640625 2
# image/rgb/20210805/src/1628156311r4yug2.jpg 1280 720
716 274 787 366 0.071875 0

 

class myIter():
    def __init__(self, file_path):
        with open(file_path, 'r')as f:
            self.lines = f.readlines()
        self.idx = 0
        self.img_len = 0
        for line in self.lines:
            if line[0] == '#':
                self.img_len += 1

    def __len__(self):
        return self.img_len
    def __iter__(self):
        return self
    def __next__(self):
        if self.idx >= len(self.lines):
            raise StopIteration
        img_name = self.lines[self.idx].split(' ')[1]
        box_list = []
        self.idx += 1
        while(self.lines[self.idx][0] != '#'):
            box_list.append(self.lines[self.idx].strip())
            self.idx += 1
            if self.idx == len(self.lines):
                return img_name, box_list
        return img_name, box_list