evaluation_script.py 文件源码

def load_images(self, test_list):

        """
            train_list : list of users to use for testing
            eg ["user_1", "user_2", "user_3"]
        """

        self.image_list = []

        for user in test_list:

            csv = "%s%s/%s_loc.csv" % (self.data_directory, user, user)

            with open(csv) as fh:
                data = [line.strip().split(',') for line in fh]

            for line in data[1:]:

                img_path, x1,y1,x2,y2, = line
                pos = tuple(map(int,(x1,y1,x2,y2)))
                letter = img_path[-6]

                img = io.imread("%s%s" % (self.data_directory, img_path))

                self.image_list.append((img, pos, letter))