davecg

#!/usr/bin/env python3

'''
This will convert Keras 1 model weights and configurations
to their Keras 2 equivalent.  Please note that all optimizer
configuration is ignored with this script and models will 
need to be recompiled.
'''

import h5py

davecg

import h5py
import shutil
import json
import sys


def fix_weight_file(model_path):
    with h5py.File(model_path, "r+") as out_h5:
        v = out_h5.attrs.get("model_config")
        config = json.loads(v)

davecg

import numpy as np
import matplotlib.pyplot as plt
import torch
import torch.nn as nn
import torch.optim as optim
import torch.nn.functional as F
from torch.autograd import Variable
import torchvision
import torchvision.transforms as transforms
import numpy as np