Advertisement
Guest User

Untitled

a guest
Mar 28th, 2017
52
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.22 KB | None | 0 0
  1. # -*- coding: utf-8 -*-
  2. """
  3. Created on Tue Mar 28 08:21:22 2017
  4.  
  5. @author: mmedek
  6. """
  7. import sys
  8.  
  9. class EvaluationData:
  10.          
  11.     def __new__(self, input_field):
  12.        
  13.          if (len(input_field) != 4):
  14.             sys.exit('In Folder \'' + input_field[0] + '\' is not three items (image.jpg, model.h5, model.json)!')
  15.                  
  16.          self.folder = input_field[0]
  17.          
  18.          if (input_field[1].rsplit('.', 1)[1] != self.__CONST_JPG()):
  19.            sys.exit('In Folder \'' + input_field[0] + '\' was not found image.jpg')
  20.          self.image_path = input_field[1]
  21.          
  22.          if (input_field[2].rsplit('.', 1)[1] != self.__CONST_H5()):
  23.              sys.exit('In Folder \'' + input_field[0] + '\' was not found h5 model file')
  24.          self.model_h5 = input_field[2]
  25.          
  26.          if (input_field[3].rsplit('.', 1)[1] != self.__CONST_JSON()):
  27.              sys.exit('In Folder \'' + input_field[0] + '\' was not found json model file')
  28.          self.model_json = input_field[3]
  29.          
  30.          return self
  31.                  
  32.     def __CONST_JPG():
  33.         return "jpg"
  34.        
  35.     def __CONST_H5():
  36.         return "h5"
  37.        
  38.     def __CONST_JSON():
  39.         return "json"
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement