深度学习标注

首先感谢万门大学的课程

1. 图像分类（利用图像名进行分类）

import random
from PIL import Image

import os
import numpy as np

################################
root='datasets/'
for dir in os.listdir(root):

    for file in os.listdir(root+dir):
        try:

            img=Image.open(root+dir+'/'+file)
            img=img.resize(100,100)      #     调整图像大小
            print(root+dir+'/'+file)
            img.save(root+dir+'/'+file)
        except:
            os.remove(root+dir+'/'+file)

################################


root='datasets/'
counter=0
for dir in os.listdir(root):
    print(dir)
    for file in os.listdir(root+dir):
        os.rename(root+dir+'/'+str(file),root+dir+'/'+str(counter)+'_'+str(file))
    counter +=1

################################
dataset_path='放文件的路径'
x=[]  # 建立一个样本列表
y=[]  #建立一个标注列表

for file in os.listdir(dataset_path):
    img=Image.open(dataset_path+file)
    img=np.array(img)
    label=int(str(file).split('_')[0])   # 基于文件名提取标签     0_******    1_****     提取出  0 和 1 
    x.append(img)
    y.append(label)

2.标注工具：精灵标注助手。