CNN Triplet Training

Instructions for training CNN triplet network with a classifier

All the necessary software scenarios for training and using the convolutional neural network model are available in our repository. Features of neural network training on image triplets helps to achieve high accuracy for small sets of training data

Clone Prject from git

git clone https://github.com/WEBSTERMASTER777/TripletRecognitionSystem.git

Load a data set, any data set with the following structure is suitable:

-- DIR WITH DATASET NAME:

    --- Dir_class_1

    --- Dir_class_2
    ...

    --- Dir class_n

The architecture of our network is great for training on small datasets, examples of datasets on kaggle:

• https://www.kaggle.com/datasets/vbookshelf/rice-leaf-diseases?resource=download
• https://www.kaggle.com/datasets/batoolabbas91/flower-photos-by-the-tensorflow-team

Install requirements:

pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html