将图片添加描述批量写入excel

原始图片

写入excel的效果

代码

python 复制代码
# by zengxy + chatgpt
# from https://blog.csdn.net/imwaters



import os
import xlsxwriter
from PIL import Image






class Image2Xlsx():

    def  __init__(self,xls_path,
                  head_list=['编号', '图片', '名称', "描述",'备注'],
                  set_default_y=112,
                  set_default_x=224
                  
                  ):

        # create a new Excel file and add a worksheet
        self.workbook = xlsxwriter.Workbook(xls_path)
        self.worksheet = self.workbook.add_worksheet()


        # title
        self.add_head(
                    headers=head_list)
        self.image_row_index = 1 # 第一行的序号为0,用来写head,所以索引从行1开始
        self.image_col_index = 1

        # 设置默认行高
        self.worksheet.set_default_row(set_default_y)
        self.fix_img_cell_x_width= set_default_x
        self.fix_img_cell_y_width = set_default_y

        # 设置 1-4列的像素宽度为224
        self.worksheet.set_column_pixels(first_col=1,last_col=4, width=self.fix_img_cell_x_width)

        
    def close(self):
        self.workbook.close()

    def add_head(
                 self,headers
                 ):
        # 设置表头的格式
        header_format = self.workbook.add_format({'bold': True, 'align': 'center'})

        # 定义表头列表

        # 在第一行循环写入表头
        for col, header in enumerate(headers):
            self.worksheet.write(0, col, header, header_format)

    def cal_img_scale(self, img_width, img_height):
        """
        计算图片的缩放比例,以便将图片适配至Excel单元格中。
        如果图片高度是宽度的1.5倍,则将单元格高度增加64像素。
        
        """
        # 设定宽高比例差异的阈值和宽度调整阈值
        aspect_ratio_thresh = 1.5
        cur_cell_width=self.fix_img_cell_y_width
        if img_width > img_height * aspect_ratio_thresh:
             cur_cell_width+= 64
    
        # 计算缩放比例,如果图片过宽,则进一步调整
        # 计算当前宽,和高度分别是原图像的倍数,以最小的为缩放比例
        scale =  min(cur_cell_width / img_height,self.fix_img_cell_x_width/img_width)
        # print('scale',scale)

        return scale

    def wirte_one_images_2_xls(self,image_path, content='by_gpt',other_des="暂时没有"):

        cur_dir,file_name=os.path.split(image_path)

        img_width, img_height = Image.open(image_path).size
        


            # 计算缩小放大比例
        scale=self.cal_img_scale( img_width, img_height)


        print("old_size:",img_width, img_height)
        print("cur_size:",int(img_width*scale), int(img_height*scale))

        '''
        写序号,第0列
        '''
        self.worksheet.write(self.image_row_index,
                        self.image_col_index - 1,
                        str(self.image_row_index))


        '''
        写入图片第2列 序号1
        '''
        
        

        self.worksheet.insert_image(self.image_row_index,
                               self.image_col_index,
                               image_path,
                               {
                                   'x_scale': scale,
                                   'y_scale': scale,
                                #    'x_offset': 5, 'y_offset': 5,
                                #    'positioning': 1  # 1 表示图片被定位为与单元格移动和大小改变而移动和改变大小。它使得图片与单元格"绑定"。
                                }
                            )
        
        '''
        写入图片名称
        '''
        self.worksheet.write(self.image_row_index,
                        self.image_col_index + 1,
                        file_name
                        )
        
        '''
        描述内容,可以是gpt生成的描述
        '''
        self.worksheet.write(self.image_row_index,
                         self.image_col_index + 2,
                        content)
        
        '''
        备注信息
        '''
        self.worksheet.write(self.image_row_index,
                         self.image_col_index + 3,
                        other_des)
        

        
        # 指向到下一行
        image2xls.image_row_index += 1


if __name__ == '__main__':

    excel_path = './test.xlsx'
    images_directory =  r'./imgaes' # Replace with the path to your images

    image2xls= Image2Xlsx(excel_path,
                          set_default_x=224,
                          set_default_y=112)

    for  file in sorted(os.listdir(images_directory)):
        img_path=os.path.join(images_directory,file )

        content="自动生成的描述,可以是gpt打标"
        other_des="这里填写你的备注信息"
        image2xls.wirte_one_images_2_xls(img_path,content,other_des)
    
    # 不关闭会报错
    image2xls.close()
相关推荐
ruleslol16 分钟前
VBA02-初识宏——EXCEL录像机
excel·vba
IT铺子16 分钟前
制定Excel使用规范和指导,提升数据处理的效率和准确性,减少错误和数据丢失的风险
excel
q5673152318 分钟前
在 Bash 中获取 Python 模块变量列
开发语言·python·bash
是萝卜干呀19 分钟前
Backend - Python 爬取网页数据并保存在Excel文件中
python·excel·table·xlwt·爬取网页数据
代码欢乐豆20 分钟前
数据采集之selenium模拟登录
python·selenium·测试工具
许野平43 分钟前
Rust: 利用 chrono 库实现日期和字符串互相转换
开发语言·后端·rust·字符串·转换·日期·chrono
也无晴也无风雨1 小时前
在JS中, 0 == [0] 吗
开发语言·javascript
狂奔solar1 小时前
yelp数据集上识别潜在的热门商家
开发语言·python
Tassel_YUE1 小时前
网络自动化04:python实现ACL匹配信息(主机与主机信息)
网络·python·自动化
聪明的墨菲特i1 小时前
Python爬虫学习
爬虫·python·学习