Skip to content

Latest commit

 

History

History
51 lines (36 loc) · 2.32 KB

File metadata and controls

51 lines (36 loc) · 2.32 KB

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