Binary classification between dogs and cats using alexnet inspired architecture
python3, numpy, matplotlib
OpenCV
CUDA 10.2 #tested with this particular version
Pytorch 1.4
- Clone the repository
- To understand the data
python3 data_statistics.py --data=<path-to-data>
- Data preperation
python3 data_prep.py --data=<path-to-data> --output=<path-to-output>
- Running the code
sh train.sh