keras to tensorflow.keras

requirements.txt

@@ -1,10 +1,10 @@
-Keras==2.1.2
-tensorflow==1.4.0
-nltk==3.2.5
-opencv-python==3.3.0.10
-numpy==1.13.1
-h5py==2.7.1
-matplotlib==2.0.2
-Pillow==4.3.0
-tqdm==4.17.1
-scipy==1.0.0
+Keras==2.6.0
+tensorflow==1.15.0
+nltk==3.6.5
+opencv-python==4.5.4.58
+numpy==1.18.5
+h5py==2.10.0
+matplotlib==3.3.4
+Pillow==8.4.0
+tqdm==4.62.3
+scipy==1.4.1

src/classes/dataset/Dataset.py

@@ -6,9 +6,9 @@
 import hashlib
 import numpy as np
 
-from keras.preprocessing.text import Tokenizer, one_hot
-from keras.preprocessing.sequence import pad_sequences
-from keras.utils import to_categorical
+from tensorflow.keras.preprocessing.text import Tokenizer, one_hot
+from tensorflow.keras.preprocessing.sequence import pad_sequences
+from tensorflow.keras.utils import to_categorical
 
 from .ImagePreprocessor import *
 

src/classes/dataset/ImagePreprocessor.py

@@ -7,7 +7,7 @@
 import numpy as np
 from PIL import Image
 import cv2
-from keras.preprocessing.image import ImageDataGenerator
+from tensorflow.keras.preprocessing.image import ImageDataGenerator
 
 class ImagePreprocessor:
 

src/classes/inference/Sampler.py

@@ -6,8 +6,8 @@
 import json
 import numpy as np
 
-from keras.models import model_from_json
-from keras.preprocessing.sequence import pad_sequences
+from tensorflow.keras.models import model_from_json
+from tensorflow.keras.preprocessing.sequence import pad_sequences
 
 from classes.dataset.Dataset import *
 from classes.dataset.ImagePreprocessor import *

src/classes/model/SketchCodeModel.py

@@ -1,11 +1,11 @@
 from __future__ import absolute_import
 
-from keras.models import Model, Sequential, model_from_json
-from keras.callbacks import ModelCheckpoint, CSVLogger, Callback
-from keras.layers.core import Dense, Dropout, Flatten
-from keras.layers import Embedding, GRU, TimeDistributed, RepeatVector, LSTM, concatenate , Input, Reshape, Dense
-from keras.layers.convolutional import Conv2D
-from keras.optimizers import RMSprop
+from tensorflow.keras.models import Model, Sequential, model_from_json
+from tensorflow.keras.callbacks import ModelCheckpoint, CSVLogger, Callback
+from tensorflow.keras.layers.core import Dense, Dropout, Flatten
+from tensorflow.keras.layers import Embedding, GRU, TimeDistributed, RepeatVector, LSTM, concatenate , Input, Reshape, Dense
+from tensorflow.keras.layers.convolutional import Conv2D
+from tensorflow.keras.optimizers import RMSprop
 
 from .ModelUtils import *
 from classes.dataset.Dataset import *