fix tile info

hanzhi713 · Apr 17, 2022 · c7f6833 · c7f6833
1 parent 7e808b9
commit c7f6833
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/make_img.py b/make_img.py
@@ -468,7 +468,7 @@ def make_photomosaic(self, assignment: np.ndarray):
         grid_assignment = assignment.reshape(self.grid[::-1])
         combined_img = self.combined_img[grid_assignment, :, : , :].transpose((0, 2, 1, 3, 4))
         return combined_img.reshape(np.prod(combined_img.shape[:2]), -1, 3), \
-               f"Grid dimension: {self.grid}\n" + '\n'.join([img.info for img in self.imgs])
+               f"Grid dimension: {self.grid}\n" + '\n'.join([self.imgs[i].info for i in assignment])
 
     def convert_colorspace(self, img: np.ndarray):
         if self.flag is None: