Awesome Open Source
Awesome Open Source

Keras vs. PyTorch: Alien vs. Predator recognition with transfer learning

Featured in deepsense.ai blog post Keras vs. PyTorch: Alien vs. Predator recognition with transfer learning, in which we discuss the differences. Code is in two Jupyter Notebooks:

See also the upcoming webinar (10 Oct 2018), in which we walk trough the code.

For plug&play interactive code, see the Neptune versions with fancy charts or these Kaggle Kernels:

Data

See also: Alien vs. Predator images | Kaggle. In general, there are 447 images for each class, split into two classes. Examples:

Requirements

If you want to run the code, see the requirements:

  • Common:
    • jupyter==1.0.0
    • matplotlib==2.2.3
    • Pillow==5.2.0
    • h5py==2.8.0
  • Keras:
    • tensorflow==1.10.1
    • Keras==2.2.2
  • PyTorch:
    • torch==0.4.1
    • torchvision==0.2.1

Webinar info


Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
jupyter-notebook (6,416
pytorch (2,448
keras (778
pytorch-implementation (89
pytorch-tutorial (70
pytorch-tutorials (24
keras-models (23
keras-neural-networks (19
keras-tutorials (19
pytorch-cnn (16