Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- """
- Created on Tue Mar 28 08:21:22 2017
- @author: mmedek
- """
- import sys
- class EvaluationData:
- def __new__(self, input_field):
- if (len(input_field) != 4):
- sys.exit('In Folder \'' + input_field[0] + '\' is not three items (image.jpg, model.h5, model.json)!')
- self.folder = input_field[0]
- if (input_field[1].rsplit('.', 1)[1] != self.__CONST_JPG()):
- sys.exit('In Folder \'' + input_field[0] + '\' was not found image.jpg')
- self.image_path = input_field[1]
- if (input_field[2].rsplit('.', 1)[1] != self.__CONST_H5()):
- sys.exit('In Folder \'' + input_field[0] + '\' was not found h5 model file')
- self.model_h5 = input_field[2]
- if (input_field[3].rsplit('.', 1)[1] != self.__CONST_JSON()):
- sys.exit('In Folder \'' + input_field[0] + '\' was not found json model file')
- self.model_json = input_field[3]
- return self
- def __CONST_JPG():
- return "jpg"
- def __CONST_H5():
- return "h5"
- def __CONST_JSON():
- return "json"
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement