This commit is contained in:
wailovet
2024-07-02 00:54:11 +08:00
parent ec6144fa81
commit 6babea03a6
+1
View File
@@ -394,6 +394,7 @@ def MZ_HYDiTTrain_call(args={}):
writer.writerow(["image_path", "text_zh"])
for filename in full_filenames:
print(f"处理文件: {filename}")
if filename.lower().endswith(".png") or filename.lower().endswith(".jpg"):
image_path = os.path.join(workspace_images_dir, filename)