hitchhicker

import _pickle as cPickle
import gzip

import numpy as np
from sklearn.utils import shuffle


def load_data():
    f = gzip.open('./data/mnist.pkl.gz', 'rb')
    training_data, validation_data, test_data = cPickle.load(f, encoding='bytes')

hitchhicker

import numpy as np

import tensorflow as tf

TEST_IMAGE_PATH = './glass17.jpg'
FROZEN_MODEL_PATH = './graph.pb'
INPUT_SIZE = 224
INDEX_2_LABEL = {
    0: 'cardboard',
    1: 'plastic',