python | 细粒度分类图像标签的一些处理操作

最新推荐文章于 2022-12-24 17:45:00 发布

努力努力再努力tq

最新推荐文章于 2022-12-24 17:45:00 发布

阅读量1k

点赞数 3

分类专栏： python

本文链接：https://blog.csdn.net/u012426298/article/details/82142049

版权

本文将介绍如何使用Python进行细粒度图像分类，包括处理图像标签等关键步骤。敬请期待详细的教程发布。

摘要由CSDN通过智能技术生成

#-*-coding:utf8-*-
import os
import  numpy as np
import sys
from PIL import Image
import string
import re
labeldict = {}
def generate_label(labeldir, traindir):
    with open(labeldir) as f:
        labellist = f.readlines()
        print(len(labellist))
        for item in labellist:
            (labelname, No) = item.strip('\n').split(',')
            labeldict[labelname] = No
        #for key in labeldict:
        #    print key, labeldict[key]
    file_list = os.listdir(traindir)
    for item in file_list:
        #print item
        pattern = re.split('\d$',item.split('_', 1)[1].rstrip(".jpg"))[0]
        #print(item.split('_', 1)[1].rstrip(".jpg"))
        if pattern in labeldict.keys():
            #print(item, pattern, labeldict[pattern])
            with open('train_data0.txt', 'a') as fw:
                 fw.write(item + " "