Image-Search-Engine

Image Search Engine implementation in PyTorch from scratch.

Training

To train our model run main.py. To change training parameters, change argumuments in constructor of class Utils when instancing an object in main.py.

Parameter	Default value	Description
EPOCHS	500	number of epochs for training
batchSize	256	size of batch during training
learning_rate	0.0005	learning rate for training
optim	Adam	optimizer used for training (SGD or Adam)
weightDecay	None	used with SGD optimizer
momentum	None	used with SGD optimizer
lastLayerActivation	Sigmoid	last layer activation function

Testing

To test our model run usemodel.py, and set path parameter to where the model is saved.

Implementation

Refer to Image-Search-Engine.pdf