SiameseNetworkPytorch

Classify people with DL-based image classification model efficientnetb0 (paper) trained using Siamese Network, test the model performance on unseen images during training.

Create virtual environment

conda create -n <ENV_NAME> python=3.9
conda activate <ENV_NAME>
pip install -r requirements.txt

Unzip Input Images

unzip data.zip

Sample data

Run training

python train.py --batch_size=64 --lr=3e-4 --model_name="efficientnet_b3a"

Results