python 图片数据集批量打标签

1.数据集介绍

OutdoorScene数据集:http://people.csail.mit.edu/torralba/code/spatialenvelope/

This dataset contains 8 outdoor scene categories: coast, mountain, forest, open country, street, inside city, tall buildings and highways.

There are 2688 color images, 256x256 pixels. All the objects and regions in this dataset have been fully labeled. There are more than 29.000 objects. The annotations are available in LabelMe format.

下载数据集并解压以后是这个样子的:

部分数据集图片

 2.python实现批量打类别标签

可以看到数据集中的图片按 类别_编号.jpg 统一命名,用sqlit切割文件名,再判断他们属于哪一类,将结果保存到矩阵中,写入csv文件中即可。

import os
import numpy as np
import csv
'''
将outdoorscene数据集中的8类图片分别打标签
directory_name:数据集路径
'''
def read_directory(directory_name):
    
    i = 0
    print('文件个数:',len(os.listdir(directory_name)))
    img_target = np.zeros((len(os.listdir(directory_name)),8),dtype=int)

    for filename in os.listdir(directory_name):
       
        category = filename.split('_',1)[0]

        if category == 'coast':
            img_target[i,0]=1
        elif category == 'forest':
            img_target[i,1]=1
        elif category == 'highway':
            img_target[i,2]=1
        elif category == 'insidecity':
            img_target[i,3]=1
        elif category == 'mountain':
            img_target[i,4]=1
        elif category == 'opencountry':
            img_target[i,5]=1
        elif category == 'street':
            img_target[i,6]=1
        elif category == 'tallbuilding':
            img_target[i,7]=1            
        
        i=i+1

    return img_target

csvFile = open('target.csv','w', newline='')
writer = csv.writer(csvFile)
writer.writerows(list(read_directory("D:\spatial_envelope_256x256_static_8outdoorcategories")))
csvFile.close()

程序执行结果:

csv文件
  • 6
    点赞
  • 99
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值