读取csv并通过url下载照片，制作数据集

最新推荐文章于 2024-06-28 13:50:39 发布

学霸的作业会繁衍

最新推荐文章于 2024-06-28 13:50:39 发布

阅读量1.1k

点赞数 1

文章标签： python

本文链接：https://blog.csdn.net/ZZZJunFa/article/details/105010855

版权

读取csv并通过url下载照片，制作数据集

首先准备一个csv文件

import csv

#打开.csv提取URL存入列表a
with open('青苹果.csv', 'r',encoding='utf-8') as f:
    a = []
    url = ""
    reader = csv.reader(f)
    for row in reader:
        print(row)
    #     a.append(row[1][12:-2])
    #     url = row[1][12:-2]
    #
    # print(a)

拥有id,oss_data,label三个key（关键字），这里print一下，row的类型是list（列表）
在这里插入图片描述
2. 接着遍历截取url并下载到本地

#下载图片到本地./image文件夹
def request_download(oss,url):
    r = requests.get(url)
    with open("./image/"+str(oss)+".png", 'wb') as f:
        f.write(r.content)

3.最后制作数据集，代码如下

def img_tra():
    for k in range(0,num):
        currentpath=folder+"/"+imglist[k]
        im=Image.open(currentpath)
        #width=im.size[0]
        #height=im.size[1]
        x_s=32
        y_s=32
        out = im.resize((x_s,y_s),Image.ANTIALIAS)
        out.save(folder_ad+"/"+str(imglist[k]))
def addWord(theIndex,word,adder):
    theIndex.setdefault(word,[]).append(adder)
def seplabel(fname):
    filestr=fname.split(".")[0]
    label=int(filestr.split("_")[0])
    return label
def mkcf():
    global data
    global list1
    global list2
    global list3
    for k in range(0,num):
        currentpath=folder_ad+"/"+imglist[k]
        im=Image.