Skip to content

RandomCyberCoder/Mahjong_Shifu_Project

Repository files navigation

Images that create the tile_images dataset are obtained from https://universe.roboflow.com/jon-chan-gnsoa/mahjong-baq4s/dataset/54 . The images used in the our mini dataset include all the test images and selected images from the train set. When selecting images from the train set we included the agumented versions of them to (rotation and rgb->grayscale).

# Our annotated dataset used to train our models
https://app.roboflow.com/mahjongshifu/mahjong_shifu/5 

# Jupyter notebooks explanation:
**active_learning.ipynb:** This has the code for our complete active learning pipline.

**augment_data.ipynb:** contains a code snippet used to manually augment images.

**project.ipynb:** contains the code we used to train and validate the model. It includes our first version of active learning. This file also includes the code used for running the live camera feed and saving a frames of the camera feed to add to our unlabeled dataset.

**Large_Model.ipynb:** contains code used to train and validate the Model Large.

**validation.ipynb:** get validation results, can ignore.

# Directory structure
**active_learning:** This directory contains four subdirectories 

* unlabled_archive: contains the all images that have been marked to for manual annotation throughout all runs of the active learning.
* unlabled_data: this directory is the dataset containing all unlabeled images
* unlabeled_labeled: contains the predictions the model made on the unlabeled dataset during an active training run.
* unlabeled_to_label: contains the images marked for manual annotation after a active learning run is completed.

**additional_images & backup:** Both of these directories can be ignored.

**runs\detect:** This directory contains all the weights of the different version of our models and also contain the validation results for each of the models. Explanation of each subdirectory:

---TRAINING--- contains training images, results metrics, and weights
* Train6_old: This is the directory for Model V1
* train: Directory for Model V2
* train2: Directory for Model V3
* train3: Directory for Model V4
* train4: Directory for a model we trained from scratch using the final version of our dataset. (This model is not included the report)
* trainLarge: Directory for Model Large
---PREDICTIONS--- : contains images annotated by our model
* predict: Directory of Model V1 prediction on the test split.
* predict2: Directory of Model V2 prediction on the test split.
* predict3: Directory of Model V3 prediction on the test split.
* predict4: Directory of Model V4 prediction on the test split.
* predictLarge: Directory of Model Large prediction on the test split.
---VALIDATIONS--- : contains visuals of the validation
* val4: Directory of Model V4 validation on the test split.
* valLarge: Directory of Model Large validation on the test split

**tile_images:** This directory can be ignored. It was originally used to store the first version of our dataset that can be found on Roboflow

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors