Abstract: Highlights•The existing capsule network (CapsNet) performs poorly on complex datasets.•GR-CapsNet uses frequential and spatial units to learn feature ignored by EM routing.•GR-CapsNet uses less GPU memory and performs inference more efficiently.•GR-CapsNet outperforms existing CapsNets on complex datasets with color backgrounds.
Loading