diff --git a/README.md b/README.md index 2002d32..47e87a7 100644 --- a/README.md +++ b/README.md @@ -105,6 +105,13 @@ display_image_keys: The repository provides several scripts for processing and analyzing ClickMe data. Below are the primary scripts along with examples using the updated configuration interface. +To get started quickly, the following example is included: `RUN.sh` + +The lines that need to be modified are as follows: +- Add your own virtual environment +- Add your own Python path + + #### Compute Human Ceiling Split Half This script performs split-half correlation analysis on the clickmaps. diff --git a/RUN.sh b/RUN.sh new file mode 100644 index 0000000..af28f07 --- /dev/null +++ b/RUN.sh @@ -0,0 +1,20 @@ +#!/bin/bash +#SBATCH --time=48:00:00 +#SBATCH -n 8 +#SBATCH -N 1 +#SBATCH --mem=128G +#SBATCH -p gpu --gres=gpu:2 +#SBATCH --account=carney-tserre-condo +#SBATCH -J Jay-ClickMe-Processing +#SBATCH -o logs/log-ClickMe-Processing-%j.out + +export PYTHONUNBUFFERED=1 +export PYTHONNOUSERSITE=1 +export PYTHONPATH="/oscar/data/tserre/jgopal/human_clickme_data_processing:$PYTHONPATH" + +source /gpfs/data/tserre/jgopal/human_clickme_data_processing/jay-venv/bin/activate + +python -u ceiling_floor_estimate.py configs/imagenet_val_oscar.yaml --save_json +python -u ceiling_floor_estimate.py configs/imagenet_train_oscar.yaml --save_json +python -u prepare_clickmaps.py configs/imagenet_val_oscar.yaml +python -u prepare_clickmaps.py configs/imagenet_train_oscar.yaml diff --git a/requirements.txt b/requirements.txt index 0b09af2..dded0fc 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,8 +1,58 @@ -numpy -pandas -Pillow -scipy -torch -matplotlib -h5py -cython \ No newline at end of file +accelerate==1.10.1 +certifi==2025.8.3 +charset-normalizer==3.4.3 +contourpy==1.3.0 +cycler==0.12.1 +Cython==3.1.3 +filelock==3.13.1 +fonttools==4.59.2 +fsspec==2024.6.1 +h5py==3.14.0 +hf-xet==1.1.9 +huggingface-hub==0.34.4 +idna==3.10 +importlib_resources==6.5.2 +Jinja2==3.1.4 +joblib==1.5.2 +kiwisolver==1.4.7 +MarkupSafe==2.1.5 +matplotlib==3.9.4 +mpmath==1.3.0 +networkx==3.2.1 +numpy==1.26.3 +nvidia-cublas-cu12==12.1.3.1 +nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cudnn-cu12==9.1.0.70 +nvidia-cufft-cu12==11.0.2.54 +nvidia-curand-cu12==10.3.2.106 +nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusparse-cu12==12.1.0.106 +nvidia-nccl-cu12==2.21.5 +nvidia-nvjitlink-cu12==12.9.86 +nvidia-nvtx-cu12==12.1.105 +packaging==25.0 +pandas==2.3.2 +pillow==11.0.0 +psutil==7.0.0 +pyparsing==3.2.3 +python-dateutil==2.9.0.post0 +pytz==2025.2 +PyYAML==6.0.2 +requests==2.32.5 +safetensors==0.6.2 +schedulefree==1.4.1 +scikit-learn==1.6.1 +scipy==1.13.1 +six==1.17.0 +sympy==1.13.1 +threadpoolctl==3.6.0 +torch==2.5.1+cu121 +torchvision==0.20.1+cu121 +tqdm==4.67.1 +triton==3.1.0 +typing_extensions==4.12.2 +tzdata==2025.2 +urllib3==2.5.0 +zipp==3.23.0 \ No newline at end of file