Open Agaspher20 opened 1 year ago
The actual problem that predict
function doesn't resize image before actually predicting.
Since in previous cells of chapter 1 notebook we trained model on images resized to 224x224 square, code of the cell should be changed to the following:
is_cat,_,probs = learn.predict(img.to_thumb(224))
print(f"Is this a cat?: {is_cat}.")
print(f"Probability it's a cat: {probs[1].item():.6f}")
hey can i work on your problem if its not solved still??
hey can i work on your problem if its not solved still??
Yes, sure
In the very beginning of chapter 1 there is a following code which tries to distinguish whether it's cat or dog:
If I run it I face the following exception:
I run this notebook in Google Colab.