Training a ResNet18 on Custom Dataset

Installation

---

• In order to install this project you need to install required packages first.

pip3 install -r requirements.txt

Run

• In order to run the project, it will download the data, trained the network, extract the features, saves the similarity matching images

python3 main.py

---

Results

1

2

3

4

5

6

7

8

9

10

Training Graphs

Wandb Training Graphs