search

gram-ai / capsule-networks

A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".
https://arxiv.org/abs/1710.09829
1.73k stars 315 forks source link

How to replace dataset and change dimensions #27

Open shrybht opened 5 years ago

shrybht commented 5 years ago

Hi, This implementation was one of the easiest to implement as well as understand. I am curious and would want to implement COCO dataset using the same code.

  1. How do I change the dimensions to 3?
  2. How to import COCO dataset into the code? I tried importing from torchvision.datasets but it is throwing an error constantly.
  3. An overview as to how to make it work on COCO dataset.

Could u please help me with this?

theaaron123 commented 5 years ago

Hi,

If you look at the svhn implementation I added and compare to the original this should give you an intuition on what needs changing for a new dataset