Investigation into the feature extraction module and lightweight network for rock image analysis | Synapse