Repo for capstone-project
Download the pre-trained model here
- You will need a unix machine
- Setup Torch by following the link
- After installing torch - we need few more packages $ luarocks install nn $ luarocks install nngraph $ luarocks install image
- To run using GPU $ luarocks install cutorch $ luarocks install cunn
- Since we are using VGGNet-16 and ResNet-18 caffee trained models we need to install loadcaffe $luarocks install loadcaffe
- To view the results $ th eval.lua -model /path/to/model -image_folder /path/to/image/directory -num_images 10
- Run the Server Class under ImageCaptioningServer
- Import the project on Android Studio
- Change the IP to your network ip
- Run the project