Przeglądaj źródła

feat: 添加处理结果收集功能,保存处理文件的状态到CSV

zhch158_admin 3 miesięcy temu
rodzic
commit
bccf773dbe
1 zmienionych plików z 11 dodań i 1 usunięć
  1. 11 1
      zhch/ppstructurev3_scheduler.py

+ 11 - 1
zhch/ppstructurev3_scheduler.py

@@ -491,7 +491,17 @@ def main():
             json.dump(final_results, f, ensure_ascii=False, indent=2)
         
         print(f"💾 Scheduler results saved to: {output_file}")
-        
+
+        # 收集文件处理结果
+        processed_files = []
+        processed_files = collect_processed_files(results)
+        output_file_processed = output_dir / f"processed_files_{args.num_processes}procs.csv"
+        with open(output_file_processed, 'w', encoding='utf-8') as f:
+            f.write("image_path,status\n")
+            for file_path, status in processed_files:
+                f.write(f"{file_path},{status}\n")
+        print(f"💾 Processed files saved to: {output_file_processed}")
+
         return 0 if successful_processes == len(results) else 1
         
     except Exception as e: