capsule-networks icon indicating copy to clipboard operation
capsule-networks copied to clipboard

Published 20 hours ago • verified publisher icon gram-ai

How to replace dataset and change dimensions

Open shrybht opened this issue 6 years ago • 1 comments

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?

shrybht avatar Feb 05 '19 18:02 shrybht

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

theaaron123 avatar Aug 15 '19 10:08 theaaron123