naturomics / CapsNet-Tensorflow

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

RGB dataset (224*224) #48

Open redhat12345 opened 6 years ago

redhat12345 commented 6 years ago

How can we use ur code in other RGB dataset? Suppose the structure of dataset is like that. it contains some sub-folder. Each sub-folder represents one class.

Class A: 0001.jpg 1 0002.jpg 1 Class B: 0001.jpg 2 0002.jpg 2

naturomics commented 6 years ago

Please read the README carefully!

If you need to apply CapsNet model to your own datasets or build up a new model with the basic block of CapsNet, please follow my new project CapsLayer

Go to my new project to get a solution. This project is not designed for other datasets

naturomics commented 6 years ago

A same issue here