python3 tensorflow alexnet训练与预测

python3 tensorflow平台下 alexnet训练与预测源码，亲测ok

import numpy as np
import os
import cv2

class Dataset:

    def __init__（self imagePath extensions）:
        self.data = createImageList（imagePath extensions）
        np.random.shuffle（self.data）
        self.num_records = len（self.data）
        self.next_record = 0

        self.labels self.inputs = zip（*self.data）

        category = np.unique（self.labels）
        self.num_labels = len（category）
        self.category2label = dict（zip（category range（len（category））））
        self.label2category = {l: k for k l in self.category2label.items（）}

        # Convert the labels to numbers
        self.labels = [self.category2label[l] for l in self.labels]

    def __len__（self）:
        return self.num_records

    def onehot（self label）:
        v = np.zeros（self.num_labels）
        v[la

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     文件        4203  2017-12-20 08:55  importData.py
     文件        3468  2016-07-07 16:47  inference.py
     文件         720  2016-07-07 16:47  predict_image.py
     文件        1088  2017-12-20 09:49  test.py
     文件        2589  2017-12-20 09:17  train.py
     文件         170  2016-07-07 16:47  README.md