taneemishere commited on
Commit
cd6f682
β€’
1 Parent(s): 4d0572e

fixing bugs

Browse files
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
__pycache__/main_program.cpython-38.pyc CHANGED
Binary files a/__pycache__/main_program.cpython-38.pyc and b/__pycache__/main_program.cpython-38.pyc differ
 
classes/model/__pycache__/pix2code2.cpython-38.pyc CHANGED
Binary files a/classes/model/__pycache__/pix2code2.cpython-38.pyc and b/classes/model/__pycache__/pix2code2.cpython-38.pyc differ
 
classes/model/autoencoder.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86d9ae3ae4c294424d3bf56f448e916893eb5374907a211d09712ec3476855b5
3
+ size 2901584
classes/model/pix2code2.py CHANGED
@@ -8,6 +8,7 @@ from keras import *
8
  from .Config import *
9
  from .AModel import *
10
  from .autoencoder_image import *
 
11
 
12
 
13
  class pix2code2(AModel):
@@ -20,7 +21,8 @@ class pix2code2(AModel):
20
  # Load the pre-trained autoencoder model
21
  autoencoder_model = autoencoder_image(input_shape, input_shape, output_path)
22
  autoencoder_model.load('autoencoder')
23
- path_to_autoencoder = "bin/autoencoder.h5"
 
24
  autoencoder_model.model.load_weights(path_to_autoencoder)
25
  # Get only the model up to the encoded part
26
  hidden_layer_model_freeze = Model(
 
8
  from .Config import *
9
  from .AModel import *
10
  from .autoencoder_image import *
11
+ import os
12
 
13
 
14
  class pix2code2(AModel):
 
21
  # Load the pre-trained autoencoder model
22
  autoencoder_model = autoencoder_image(input_shape, input_shape, output_path)
23
  autoencoder_model.load('autoencoder')
24
+ path = "./bin/"
25
+ path_to_autoencoder = "{}autoencoder.h5".format(path)
26
  autoencoder_model.model.load_weights(path_to_autoencoder)
27
  # Get only the model up to the encoded part
28
  hidden_layer_model_freeze = Model(