search

naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules
Apache License 2.0
3.8k stars 1.17k forks source link

a quesion #81

Open NilakshanKunananthaseelan opened 5 years ago

NilakshanKunananthaseelan commented 5 years ago
  1. Y = valY[:num_val_batch * cfg.batch_size].reshape((-1, 1))

what is the use of this line in main.py? 2.Can we use the code for RGB image dataset? 3.What is the use of channel parameter in CapsNet function?

AnushaMehta commented 5 years ago

@NilakshanKunananthaseelan Not very sure about your first answer, but I guess I can answer you the other two. Ans. 2 : Yes, we can use RGB image dataset by changing the load_data() function for respective RGB dataset. Ans. 3 : Mostly, The channel parameter defines the image type whether the training images are grayscale or RGB. For grayscale image, it set to 1 and for RGB, it set to 3. I am a newbie to capsule network so I may wrong. Regards.