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

Question

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

比如他的标签是这样的

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

Answer 1

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

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')

大致上就是这样，望采纳！谢谢