Okay would you please tell me what to do to show result = No snake present in the picture.
Lyk if the probability distribution is less than 50% it will show no snake present in the picture???
I need help regarding this.
Not sure if this is what you are looking for here is an example of how to use probabilities when getting predictions:
def dog_cat(img_path):
img = open_image(img_path) prediction, indice, losses = learn.predict(img) #Calculate propabilities preds_sorted, idxs = losses.sort(descending=True) pred_1_prob = np.round(100*preds_sorted[0].item(),2) if pred_1_prob <= 90: result = (f' Model IS NOT Confident: Highest Probability: ({pred_1_prob}%)') img.show() else: result = (f' Model IS confident: ({pred_1_prob}%) Prediction: {prediction}') img.show() return result
Bear in mind though that you can still get inaccurate predictions as shown in the notebook
You can access the notebook here: fastai_resources/Dog_Cat_Probabilites.ipynb at master · asvcode/fastai_resources · GitHub
1 Like