def test_recognition_model_mobilenetv2(): predictor = ImageClassification() predictor.setModelTypeAsMobileNetV2() predictor.setModelPath(os.path.join(main_folder, "data-models", "mobilenet_v2.h5")) predictor.loadModel() predictions, probabilities = predictor.classifyImage(image_input=os.path.join(main_folder, main_folder, "data-images", "1.jpg")) assert isinstance(predictions, list) assert isinstance(probabilities, list) assert isinstance(predictions[0], str) assert isinstance(probabilities[0], float)
import os from imageai.Classification import ImageClassification # get current directory execution_path = os.getcwd() prediction = ImageClassification() prediction.setModelTypeAsMobileNetV2() prediction.setModelPath(os.path.join(execution_path, "mobilenet_v2.h5")) prediction.loadModel() predictions, probabilities = prediction.classifyImage(os.path.join( execution_path, "giraffe.jpg"), result_count=5) for eachPrediction, eachProbability in zip(predictions, probabilities): print(eachPrediction, " : ", eachProbability)