Skip to content

Commit

Permalink
Merge pull request OlafenwaMoses#273 from moethu/feature/access-model…
Browse files Browse the repository at this point in the history
…-collection

OlafenwaMoses#1 get model collection
  • Loading branch information
OlafenwaMoses authored Aug 14, 2019
2 parents ecde53b + b3dd0fb commit 704223a
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions imageai/Prediction/Custom/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -351,6 +351,13 @@ def __init__(self):
self.__model_collection = []
self.__input_image_size = 224

def getModels(self):
"""
'getModels()' provides access to the internal model collection. Helpful if models are used down the line with tools like lime.
:return:
"""
return self.__model_collection

def setModelPath(self, model_path):
"""
'setModelPath()' function is required and is used to set the file path to the model adopted from the list of the
Expand Down

0 comments on commit 704223a

Please sign in to comment.