Please see the abstract here.
Models:
Transfer leaning model
YOLO model
Visualization:
Automatic Detection and Classification of Rock Microstructures through Machine Learning
All of the models and visualization are in notebooks folders. Please see the instruction below to use.
We provide requirements.txt for usage together with Binder, as well as exact replica of conda environment in env.yml
git clone https://github.com/drizzle98/sigma-clast-projectGo to your cloned directory
cd ../github/sigma-clast-projectInstall the required packages Using pip:
pip install -r requirements.txtUsing conda:
conda env create -f env.ymlconda activate microstructuresGo to notebook folder:
cd notebooksRun Jupyter lab:
jupyter labFor YOLO model specifically, please also download the support material inside the folder YOLO_support_materials.