Subversion Repositories keras_to_fpga

from __future__ import print_function

import keras

from tensorflow.keras.datasets import mnist

from tensorflow.keras.models import Sequential

from tensorflow.keras.layers import Dense, Dropout

from tensorflow.keras.models import Model

from tensorflow.keras.models import load_model

import numpy as np

import os

import time

import sys

import shutil

# -------------------------------------------------------

sys.path.insert(0, '../../scripts')

import qaz_util as qaz

# # -------------------------------------------------------

# (x_train, y_train),(x_test, y_test) = qaz.load_mnist()

# -------------------------------------------------------

model = load_model('mnist_mlp.h5')

model.summary()

# -------------------------------------------------------

dir = 'weights'

if os.path.exists(dir):

    shutil.rmtree(dir)

os.makedirs(dir)

for i in range(len(model.layers)):

  layer = model.layers[i]

  if layer.name.startswith('dense'):

    qaz.write_dense_weights(layer)