Skip to content

Repository for implementation of SalNet360 in Caffe

License

Notifications You must be signed in to change notification settings

V-Sense/salnet360

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

salnet360

Repository for implementation of salnet360 in Caffe.

salnet360 creates a visual attention map for omni-directional/360 images.

For more details on the model refer to "SalNet360: Saliency maps for omni-directional images with CNN"

Model and Parameter Files

The model and parameters files in models directory.

vsenseSalNet360Deploy.prototxt is the file that defines the network model and vsenseSalNet360Final_iter_12000.caffemodel has the optimized parameters we found using the training method mentioned in the paper.

For creating a saliency map for omni-directional image use the following command

createSalMap360.sh <inputImage> <outputImage>

Dependencies

  • Python2.7
  • Caffe
  • matlab