预训练网络实战之特征提取

        最近在做毕业设计，其中涉及到使用预训练网络的内容，网上关于这方面的资源也有很多，但是有一些代码有问题，经过一些尝试本小白还是成功的完成了这个功能，主要利用的keras库，关于keras库安装直接在命令行窗口使用pip inatall keras就可以了。在这里把我的代码发出来，望与各位同学一起进步！小白的代码写的比较渣……

# -*- coding: utf-8 -*-
"""
Created on Thu Dec 27 15:30:38 2018

@author: 13260
"""

"""
功能：提取VGG19卷积层特征并保存
"""
#from keras.applications.resnet50 import ResNet50
from keras.applications.densenet import DenseNet201,preprocess_input
from keras.preprocessing import image
#from keras.applications.resnet50 import preprocess_input
from keras.models import Model
from sklearn.cluster import MiniBatchKMeans
from keras.utils import plot_model
from sklearn import preprocessing
import numpy as np
import os

"""
功能：读取所有图片路径
输入：数据库路径
输出：所有图片路径
"""
def read_all_imgs(data_dir):
    catalogs = os.listdir(data_dir)
    imgSet = []
    for i in range(len(cata