CapsNet-Tensorflow
CapsNet-Tensorflow copied to clipboard
a quesion
- 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.