遍历图像,将有同一个标签的图像写到一个txt中

我想达到以下的效果该怎么办?比如一个图像含有多个标签

img

比如他的标签是这样的

img


我怎么能把图像数据库中含有香烟标签的图像名称全写到一个txt中

图片名和标签名是怎么对应的?

import os
path = r'D/tags'
tags = os.listdir(path)
with open ('save_cigarette_image.txt','a') as fs:
    for tag in tags:
        with open(path + '/' +tag, 'r') as f:
            if 'cigarette' in f.readlines():
                fs.write(IMAGENAME + '.jpg有animal标签\n')


大致上就是这样,望采纳!谢谢