I ultimately used Yolov5 for object detection, but I do have a second question.
I was able to use PL for image recognition using some of the examples from the blog.
I am running into an issue of generating probabilities from the exported model. I used the following code in python:
model = keras.models.load_model(path_to_model)
im = Image.open(test_image_path + test_df[‘File Name’][i])
im = np.array(im, dtype=np.float32)
im= np.expand_dims(im, axis=0)
Makes a prediction
prediction1 = model.predict(im)
model.predict only provides the class. I would like the probability of the selected identified class. Maybe you have some recommendations.