创建图像TFRecord时遇两类问题，求技术排查方案

我来帮你一步步分析这两个版本脚本遇到的问题，给出针对性的排查方向：

版本1：硬编码单张图像参数，生成TFRecord但报SystemExit异常 #

虽然成功生成了目标TFRecord，但出现的SystemExit异常大概率是以下原因导致的：

• 你的脚本里可能调用了sys.exit()语句，而如果是在**交互式环境（比如Jupyter Notebook/IPython）**中运行脚本，这个调用会触发该异常（交互式环境不允许直接退出进程），但核心的TFRecord生成逻辑其实已经执行完成了。
• 也有可能是脚本里的某个判断分支触发了sys.exit()，但没有影响到图像写入TFRecord的核心流程。

解决建议： #

1. 执行%tb命令查看完整的回溯信息，定位异常触发的具体代码行，确认是不是sys.exit()导致的。
2. 如果是交互式环境运行，建议把脚本逻辑封装成函数，去掉不必要的sys.exit()调用；如果是必须退出的场景，可以换成raise Exception()来抛出明确的错误，而非直接终止进程。
3. 验证生成的TFRecord是否可用：用tf.data.TFRecordDataset读取文件，检查图像数据是否正常解析。

版本2：批量传入5张图像参数，流程未完成 #

这种“流程卡住/未完成”的情况，通常和参数处理、文件读取或异常未捕获有关，你可以按以下步骤排查：

排查步骤： #

• 先做最小化测试：用版本2的函数单独处理1张图像（和版本1同一张），看是否能成功生成TFRecord。如果单张也失败，说明函数本身的逻辑有问题；如果单张成功，再逐步增加到5张，定位是哪张图像导致的问题。
• 验证所有输入参数的有效性：
  • 逐个检查5张图像的路径，用os.path.exists(image_path)确认文件是否存在，有没有拼写错误、相对路径/绝对路径混淆的问题。
  • 确认传入的高度、宽度和图像实际尺寸是否一致——如果参数和实际图像尺寸不匹配，在编码时可能会出现静默失败或异常中断。
• 添加日志和异常捕获：在函数里加入打印语句，比如每处理一张图像就输出Processing image X: {image_path}，看流程卡在了哪一步；同时用try-except包裹图像读取、编码的逻辑，捕获并打印异常信息：

  for idx, (path, h, w) in enumerate(zip(image_paths, heights, widths)):
      try:
          print(f"Processing image {idx+1}/{len(image_paths)}: {path}")
          # 图像读取、编码、写入TFRecord的逻辑
      except Exception as e:
          print(f"Error processing {path}: {str(e)}")
          continue
  

• 检查写入权限和输出路径：确认TFRecord的输出目录是否存在，当前用户是否有写入权限——如果路径不存在或无权限，脚本会静默失败。
• 考虑自动获取图像尺寸：没必要硬编码高度和宽度，可以在读取图像时自动获取：

  from PIL import Image
  with Image.open(image_path) as img:
      actual_h, actual_w = img.size[1], img.size[0]  # PIL返回的是(width, height)
  

内容的提问来源于stack exchange，提问作者Sethu N