Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
@Snehal-Reddy
Snehal-Reddy authored Jun 1, 2019
1 parent 15adcf8 commit edf61eb
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ Inception Modules with Connected Local and Global Features implemented with CAM.
ILGNet is a novel neural network to solve the challenging problem of aesthetic image classification. The orininal architecture is given -
![alt text](https://github.com/Snehal-Reddy/ILGNET-ActivationMaps/blob/master/Screenshot%20from%202019-06-01%2002-41-48.png)

We added used class activation methods to predict which region is responsible for aesthetic image classification. To suit the needs of CAMs we modified the network to make it better.
We added used class activation methods to predict which region is responsible for aesthetic image classification. To suit the needs of CAMs we modified the network to make it better. We trained this network on AVA dataset and the weights are stores in a .pth file.
![alt text](https://github.com/Snehal-Reddy/ILGNET-ActivationMaps/blob/master/imageedit_34_4118151269.gif)

An example of the output-
Expand All @@ -14,3 +14,4 @@ Original image -

Output Heat map + image overlay -
![alt text](https://github.com/Snehal-Reddy/ILGNET-ActivationMaps/blob/master/cam3.jpg)

0 comments on commit edf61eb

Please sign in to comment.