CapsNet-Tensorflow icon indicating copy to clipboard operation
CapsNet-Tensorflow copied to clipboard

a quesion

Open NilakshanKunananthaseelan opened this issue 7 years ago • 1 comments

  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?

@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.

AnushaMehta avatar Apr 07 '19 14:04 AnushaMehta