CapsLayer
CapsLayer copied to clipboard
CapsLayer: An advanced library for capsule theory
Hello! I've found a performance issue in /capslayer/data/datasets: `batch()` should be called before `map()`, which could make your program more efficient. Here is [the tensorflow document](https://tensorflow.google.cn/guide/data_performance?hl=zh_cn#vectorized_mapping) to support it. Detailed...
Hello,I found a performance issue in the definition of `__call__(self, batch_size, mode)` , capslayer/data/datasets/cifar10/reader.py, [dataset = dataset.map(parse_fun)](https://github.com/naturomics/CapsLayer/blob/aed1feadfe9244ff7c8fddeb26f67146ea638d2a/capslayer/data/datasets/cifar10/reader.py#L77) was called without **num_parallel_calls**. I think it will increase the efficiency of your...
Thank you for this project. Is it possible to do a multi-label classification with CapsNet such that the softmax output can predict multiple classes for each input image? Thanks so...
Hi, first of all, thanks for this great project! I noticed that in `models/main.py` (line 206) the fashion mnist dataset is called 'fashion-mnist'. ``` # Deciding which dataset to use...
Hello naturomics! Thank you for this great code that is helping me understand capsule networks. In the documentation i saw that all of the capsules are seeing activation, but when...
tensorflow2.0 how to write config.py? AttributeError: module 'tensorflow' has no attribute 'app'
Using this implementation, has anyone managed to get close to the results reported in the paper on the smallNORB dataset?
Hi, I created the datareader and datawriter according to my own dataset by following the tutorials. I made the necessary changes to config.py and main.py files. But I get an...