Upload cogagent_infer_batch.py with huggingface_hub

cogagent_infer_batch.py

@@ -146,7 +146,7 @@ def main():
         img_path,task = x
         image = []
         for path in img_path:
-            image = Image.open(path).convert("RGB")
+            image.append(Image.open(path).convert("RGB"))
         #image = Image.open(img_path).convert("RGB")
         #task = x['conversations'][0]['value']
         # Verify history lengths match
@@ -204,10 +204,19 @@ def main():
             boxes = [[int(x) / 1000 for x in match] for match in matches]
 
             # Extract base name of the user's input image (without extension)
-            base_name = os.path.splitext(os.path.basename(img_path))[0]
+            base_name = []
+            for path in args.img_path:
+                base_name.append(os.path.splitext(os.path.basename(path))[0])
+            #base_name = os.path.splitext(os.path.basename(img_path))[0]
             # Construct the output file name with round number
-            output_file_name = f"{base_name}_{round_num}.png"
-            output_path = os.path.join(args.output_image_path, output_file_name)
+            output_file_name = []
+            for i in range(len(base_name)):
+                output_file_name.append(f"{base_name[i]}_{round_num}_{i}.png")
+            #output_file_name = f"{base_name}_{round_num}.png"
+            output_path = []
+            for x in output_file_name:
+                output_path.append(os.path.join(args.output_image_path, x))
+            #output_path = os.path.join(args.output_image_path, output_file_name)
 
             draw_boxes_on_image(image, boxes, output_path)
             #print(f"Annotated image saved at: {output_path}")