https://github.com/cloudinary/dl_salicon_exercise
an example application
https://github.com/cloudinary/dl_salicon_exercise
Last synced: 5 days ago
JSON representation
an example application
- Host: GitHub
- URL: https://github.com/cloudinary/dl_salicon_exercise
- Owner: cloudinary
- License: mit
- Created: 2024-08-07T11:38:33.000Z (almost 2 years ago)
- Default Branch: main
- Last Pushed: 2024-08-08T18:49:22.000Z (almost 2 years ago)
- Last Synced: 2024-08-09T21:34:23.774Z (almost 2 years ago)
- Language: Python
- Size: 13.7 KB
- Stars: 0
- Watchers: 13
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
## SALICON saliency prediction
### Get the data
from the provided S3 link
### Download the code
git clone https://github.com/cloudinary/dl_salicon_exercise.git
### Build and run the docker
```
cd dl_salicon_exercise
export DATA_DIR=[parent dir of of the SALICON dataset]
export HOST_WORKDIR=$(pwd)
docker compose -f docker-compose.yml build
docker compose -f docker-compose.yml run salicon
```
### Run training
```
python train.py --config_file train_config.json --data_path /home/data/SALICON --output_path ./output
```
### Evaluate crops
```
python evaluate.py --config_file train_config.json --data_path /home/data/SALICON --weight_file ./output/salicon_model.pth --output_path ./evaluate_output
```
### Licence
Released under the MIT license.