Sean/CNN
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix directory issue

Browse Source
This commit is contained in:
Sean
2024-10-28 10:54:36 -04:00
parent d263ef3ff8
commit b3cf821b0e
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
Models/model_host.py
View File

@@ -157,12 +157,16 @@ def process_image():
if __name__ == '__main__':
path = os.getcwd()
resnet50_model_name='Weights/resnet50.h5.keras'
if(not path.endswith('Models')):
os.chdir(path + '/Weights')
path=os.getcwd()
print('Current directory is ..{current_directory}'.format(current_directory=path))
resnet50_model_name='../Weights/resnet50.h5.keras'
resnet50_model_name = os.path.realpath(path + '/' +resnet50_model_name)
print('Loading {model_name}'.format(model_name=resnet50_model_name))
resnet50_model = keras.models.load_model(resnet50_model_name)
resnet50_20241024_270_model_name='Weights/resnet50_20241024_270.h5.keras'
resnet50_20241024_270_model_name='../Weights/resnet50_20241024_270.h5.keras'
resnet50_20241024_270_model_name = os.path.realpath(path + '/' + resnet50_20241024_270_model_name)
print('Loading {model_name}'.format(model_name=resnet50_20241024_270_model_name))
resnet50_20241024_270_model = keras.models.load_model(resnet50_20241024_270_model_name)
@@ -171,12 +175,12 @@ if __name__ == '__main__':
# print('Loading {model_name}'.format(model_name=resnet50b_model_name))
# resnet50b_model = keras.models.load_model(resnet50b_model_name)
vgg16_model_name='Weights/vggnet16.h5.keras'
vgg16_model_name='../Weights/vggnet16.h5.keras'
vgg16_model_name = os.path.realpath(path + '/' + vgg16_model_name)
print('Loading {model_name}'.format(model_name=vgg16_model_name))
vgg16_model=keras.models.load_model(vgg16_model_name)
lenet_model_name='Weights/lenet5.h5.keras'
lenet_model_name='../Weights/lenet5.h5.keras'
lenet_model_name = os.path.realpath(path + '/' +lenet_model_name)
print('Loading {model_name}'.format(model_name=lenet_model_name))
lenet_model=keras.models.load_model(lenet_model_name)