You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Finetuning AlexNet's last classifier layer where the classes are reduced from 1000 to the number of classes in dataset(8).
6
+
7
+
**Installing Dependencies**
8
+
9
+
1. Install PyTorch by following the instructions given on the link [here](http://pytorch.org/).
10
+
2. Run ```pip3 install torchvision tensorflow==1.4 tensorboardX``` in command line.
11
+
12
+
**Running the Code**
13
+
14
+
1. First, get the dataset in two folders: train and test by modifying paths in [data_process.py](https://github.com/radonys/CV-Assignments/blob/master/Assignment-2/Q3/data_process.py).
15
+
Then, execute the command ```python3 data_process.py```.
16
+
2. Then modify the dataset path in [alexnet.py](https://github.com/radonys/CV-Assignments/blob/master/Assignment-2/Q3/alexnet.py), where the newly created folders are saved.
17
+
3. Then run ```python3 alexnet.py```.
18
+
19
+
**Visualizing the Results**
20
+
21
+
To visualize the current results, open a terminal session, and run the following command:
0 commit comments