最近工作需要，写了如题的python脚本，只要安装了python3即可使用。
我的python环境
Windows + python 3.9.6，64位
代码逻辑
图片遍历转存base64工具 代码结构.png
可以下载慢慢看：链接：https://pan.baidu.com/s/13nUjXeh7igx9Q3gGzYLVyg?pwd=1dkd
提取码：1dkd
工具说明
递归遍历指定文件夹下所有图片（支持多层级文件夹），转为base64，存入指定文件夹下的txt文件中。
- 代码中判断条件为“.jpg”。(可修改代码，增改格式要求)
- 每个base64存一行。
- txt文件命名规则：1.txt、2.txt……依次命名。
- img_root参数：
（必填，图片根目录） - batch参数：
（非必填，每个txt最多存储batch条base64，超出自动新建txt并写入）
默认为0——txt存储数据数量无限制，所有数据写入1.txt；
任意正整数n——超出n条，新建txt文件并写入。 - txt_root参数：
（非必填，txt文件位置）
空——自动在img_root下创建文件夹“img_base64_txt”；
非空但不存在：自动创建该目录。
效果展示
image.png
企业微信截图_20220521182813.jpg
企业微信截图_20220521183013.jpg
企业微信截图_20220521223143.jpg
企业微信截图_20220521183049.jpg
代码
完整代码如下，粘贴到编辑器即可使用，代码最下方有示例：
# -*- coding: utf-8 -*-
import base64
import datetime
import math
import os
"""
遍历指定目录下的所有图片，存储 “filename,图片base64码”至txt文件。如“1.jpg,base64字符串”
文件名从1.txt开始，依次递增
参数：
img_root：图片文件夹路径。注意：末尾不加斜杠。例：C:\zhangyue1\桌面\测试数据\人像
batch：每个文件存放的数据行数，超出则创建新的txt文件继续写入
txt_root：存放txt文件的目录，如无，则自动创建该目录
"""
# 图片转base64
def img2base64(img_path_str):
"""图片转base64并返回str
:param img_path_str: 图片完整路径str，如r"C:\zhangyue1\桌面\测试数据\人像\1.jpg"
:return: 返回转码后的base64字符串
"""
with open(img_path_str, 'rb') as f:
img_path_str = f.read()
image_base64 = str(base64.b64encode(img_path_str), encoding='utf-8')
return image_base64
# 将str数据data追加存入文件file_path中。
def save_data(file_path: str, data: str):
"""将str数据data追加存入文件file_path中。
:param file_path: 文件完整路径。如“D:/test/1.txt”
:param data: 数据字符串
:return: 无返回值
"""
# mode='a'，追加模式：不清空文档已有数据，每次在文档末尾追加data
with open(file=file_path, mode='a', encoding="utf-8") as file:
file.write(data)
# 遍历文件夹img_root中所有图片，转base64，逐行写到txt_root文件夹中的txt文件中,可设置每个txt文件写入个数（每个base64为1行）
def find_all_img_2_txt(img_root: str, txt_root: str, batch: int = 32):
"""
:param img_root: (必填)str，图片根目录
:param txt_root: (非必填)txt文件存放位置。
空，自动在img_root下创建文件夹“img_base64_txt”；
非空但不存在，自动创建该目录。
:param batch: 每个文件存储的图片数据个数（1个图存1行）。
默认为0，全部存入1.txt；
1-n，每个文件存batch条
:return: (str) result/error，成功后统计信息/错误提示
"""
''' 一、初始化 '''
# 记录运行开始时间
start_time = datetime.datetime.now()
# 已写入文件的图片总数量。初始为0-未写入
image_number = 0
# 图片存入的txt文件名称序号，初始为0-未写入
txt_name = 0
# 判断：img_root是否存在。不存在，print错误提示并返回
if img_root is None or img_root == "" or not os.path.exists(img_root):
error = f"图片路径\"{img_root}\"不存在"
print(error)
return error
# 判断：txt_root为空。默认在img_root下创建名为“img_base64_txt”的文件夹
if txt_root is None or txt_root == "":
txt_root = os.path.join(img_root, "img_base64_txt")
os.makedirs(txt_root)
# 判断：txt_root是否存在。不存在，创建该目录
if not os.path.exists(txt_root):
os.makedirs(txt_root)
''' 二、执行 '''
# 3种情况：
# 1、batch==0，全部存到1.txt中
if batch == 0:
# 1-1.写入文件的序号固定为1
txt_name = 1
# 1-2.拼文件存储path。使用变量txt_name替换"{txt_name}"
data_path = str(os.path.join(txt_root, f"{txt_name}.txt"))
# 1-3.递归遍历图片路径（os.walk(path)：按照目录层级path，逐层遍历所有文件夹dirs、文件files。path下包含：dirs列表+files列表）
all_in_img_root = os.walk(img_root)
# 1-4.遍历所有文件
for path, dirs, files in all_in_img_root:
for filename in files:
# 1-4-1.找jpg文件。判断：文件名后缀是否“.jpg”。是，说明是图片，处理；否，不是图片，遍历下一个。
if filename.endswith(".jpg"):
# ① 图片计数+1
image_number += 1
# ② 图片转base64
img_base64_str = img2base64(os.path.join(path, filename))
# ③ base64写入1.txt：第1条数据，开头不用换行；以后每条数据，换行+数据字符串
if image_number == 1:
save_data(data_path, img_base64_str)
else:
save_data(data_path, "\n" + img_base64_str)
# 1-5.遍历完成后，判断：写入图片数量。0-没有，print提示
if image_number == 0:
error = f"文件夹\"{img_root}\"中，没有图片。"
print(error)
return error
# 2、batch>0，根据已写入图片数量，判断是否要写入新的txt文件
elif batch > 0:
# 2-1.递归遍历图片路径
all_in_img_root = os.walk(img_root)
# 2-2.遍历当前所有文件
for path, dirs, files in all_in_img_root:
for filename in files:
# 2-2-1.找jpg文件。判断：文件名后缀是否“.jpg”。是，说明是图片，处理；否，不是图片，遍历下一个。
if filename.endswith(".jpg"):
# ① 图片计数+1
image_number += 1
# ② 图片转base64
img_base64_str = img2base64(os.path.join(path, filename))
# ③ 判断：image_number除以batch的余数
# 余数=1：txt_name自增，开头不换行
if image_number % batch == 1:
txt_name += 1
# 拼文件存储path。使用变量txt_name替换"{txt_name}"
data_path = str(os.path.join(txt_root, f"{txt_name}.txt"))
save_data(data_path, img_base64_str)
# 余数!=1：txt_name不自增，开头换行
if image_number % batch != 1:
# 拼文件存储path。使用变量txt_name替换"{txt_name}"
data_path = str(os.path.join(txt_root, f"{txt_name}.txt"))
save_data(data_path, "\n" + img_base64_str)
# 1-5.遍历完成后，判断：写入图片数量。0-没有，print提示
if image_number == 0:
error = f"文件夹\"{img_root}\"中，没有图片。"
print(error)
return error
# 3、batch为其他-异常，print错误提示
else:
error = f"batch参数：{batch}，错误！请输入大于等于0的整数。"
print(error)
return error
''' 成功结束后统计 '''
# 记录运行结束时间
end_time = datetime.datetime.now()
#
latency = end_time - start_time
result = r"目标：遍历图片 + 转base64 + 存入txt文件。已完成！" + "\n\n" + \
r"============ 结果统计 ============" + "\n" + \
f"转存图片数量：{image_number}" + "\n" + \
f"txt文件数量：{txt_name}" + "\n" + \
f"最大序号txt文件名：{txt_name}.txt" + "\n" + \
f"txt文件容量最大值：{batch}" + "\n" + \
f"图片根目录：{img_root}" + "\n" + \
f"txt文件位置：{txt_root}" + "\n" + \
f"总耗时：{latency}" + "\n" + \
r"============ 结果统计 ============"
print(result)
return result
if __name__ == '__main__':
# 图片根目录
img_root_str = r"D:\image"
# 文件存储根目录
txt_root_str = r"D:\list"
find_all_img_2_txt(img_root=img_root_str, txt_root=txt_root_str, batch=10)
