readme: ---
  tags:
  - image-classification
  widget:
  - src: https://upload.wikimedia.org/wikipedia/commons/0/0c/About_The_Dog.jpg
    example_title: Dog-1
  - src: https://yt3.ggpht.com/ytc/AKedOLRvxGYSdEHqu0X4EYcJ2kq7BttRKBNpfwdHJf3FSg=s900-c-k-c0x00ffffff-no-rj
    example_title: Dog-2
  - src: https://upload.wikimedia.org/wikipedia/commons/c/c7/Tabby_cat_with_blue_eyes-3336579.jpg
    example_title: Cat-1
  - src: https://pixabay.com/get/g31cf3b945cf9b9144eb6c1ecf514b4db668875b75d0c615e0330aec74bef5edde11567ef4a6f5fdb61a828b8086a39d3a0e72fb326d78467786dcdde4e6fa23c5c4c309d0abc089a8663809c175aee22_1920.jpg
    example_title: Cat-2
  ---
  
  # Classify Cats and Dogs
  
  VGG16 fine tuned to classify cats and dogs
  
  Notebook
  
  https://www.kaggle.com/carlosaguayo/cats-vs-dogs-transfer-learning-pre-trained-vgg16
  
  ### How to use
  
  Here is how to use this model to classify an image as a cat or dog:
  
  ```python
  from skimage import io
  import cv2
  import matplotlib.pyplot as plt
  from huggingface_hub import from_pretrained_keras
  %matplotlib inline 
  
  ROWS, COLS = 150, 150
  
  model = from_pretrained_keras("carlosaguayo/cats_vs_dogs")
  
  img_url = 'https://upload.wikimedia.org/wikipedia/commons/0/0c/About_The_Dog.jpg'
  # img_url = 'https://upload.wikimedia.org/wikipedia/commons/c/c7/Tabby_cat_with_blue_eyes-3336579.jpg'
  
  img = io.imread(img_url)
  img = cv2.resize(img, (ROWS, COLS), interpolation=cv2.INTER_CUBIC)
  img = img / 255.0
  img = img.reshape(1,ROWS,COLS,3)
  
  prediction = model.predict(img)[0][0]
  if prediction >= 0.5:
      print('I am {:.2%} sure this is a Cat'.format(prediction))
  else: 
      print('I am {:.2%} sure this is a Dog'.format(1-prediction))
          
  plt.imshow(img[0], 'Blues')
  plt.axis("off")
  plt.show()
  ```