Commit 2e8343b8 authored by Boris Lukačovič's avatar Boris Lukačovič
Browse files

ADD: script to init datasets

parent 7db1a664
Loading
Loading
Loading
Loading

datasets/init.sh

0 → 100644
+41 −0
Original line number Diff line number Diff line
#!/bin/bash

# --------------------------------------- #
# ---> Dataset initialisation script <--- #
# --------------------------------------- #

folder_check() {
    curr_folder=$(basename $(pwd))
    if [[ "$curr_folder" == "datasets" ]]; then
        return 0
    fi
    echo Please run this script from inside the ./datasets folder
    exit 1
}

laion2B() {
    echo Download for laion2D is to be implemented
}

laion2B_clustered() {
    echo Download for laion2B_clustered is to be implemented
}

sift() {
    mkdir -p sift
    cd sift
    file_name=sift-128-euclidean.hdf5
    if [[ -f "$file_name" ]]; then
        echo Sift dataset is already present. Skipping download
        return 0
    fi
    wget 'http://ann-benchmarks.com/sift-128-euclidean.hdf5'
    echo Download of sift dataset completed
    cd ..
}

folder_check

laion2B
laion2B_clustered
sift
 No newline at end of file