Fix image list mismatch (#1579)

Co-authored-by: Miro <[email protected]>

text_to_image/coco.py

@@ -174,9 +174,9 @@ def save_images(self, ids, ds):
         for id in ids:
             caption = ds.get_caption(id)
             generated = Image.fromarray(self.results[idx[id]])
-            image_path_tmp = f"images/{self.content_ids[id]}.png"
+            image_path_tmp = f"images/{self.content_ids[idx[id]]}.png"
             generated.save(image_path_tmp)
-            info.append((self.content_ids[id], caption))
+            info.append((self.content_ids[idx[id]], caption))
         with open("images/captions.txt", "w+") as f:
             for id, caption in info:
                 f.write(f"{id}  {caption}\n")

tools/submission/submission_checker.py

@@ -1785,6 +1785,8 @@ def check_extra_files(path, target_files):
                 if target_file not in files:
                     check_pass = False
                     missing_files.append(f"{os.path.join(path, dir, target_file)}.png")
+            if "captions" not in files:
+                missing_files.append(f"{os.path.join(path, dir, 'captions.txt')}")
     return check_pass, missing_files