/MaSeResNeXt

the paper of MaSeResNeXt

Primary LanguagePythonApache License 2.0Apache-2.0

Python scripts used to perform rock thin section classification using MaSeResNeXt convolutional neural networks as was published in:

Rock thin sections Identification based on improved Squeeze-and-Excitation Networks model

Journal title: Computers and Geosciences Authors: He Ma; Long Peng; Liying Zhu, Jin Shu. Corresponding author: Guoqing Han

Data

Data can be downloaded here

Scripts

  • maseresnext.py: network model
  • train.py: training program
  • eval.py: evaluation model
  • infer.py: image infer
  • reader.py: image preprocessing
  • config.py: model configuration
  • utils.py: Log and other common tool functions