For CPU Installation
pip install -r requirements.txtFor GPU installlation
pip install -r requirements_gpu.txt
usage: main.py [-h] [-i INPUT] [-c CLUSTER] [-p PCA]
Image caption CLI
optional arguments:
-h, --help show this help message and exit
-i INPUT, --input INPUT Input directory path, such as ./images
-c CLUSTER, --cluster CLUSTER How many cluster will be
-p PCA, --pca PCA PCA Dimensions
--cpu Run on CPUFor CPU
python main.py -i "D:\New folder\random_1500" -c 5 -p 16 --cpuFor GPU
python main.py -i "D:\New folder\random_1500" -c 5 -p 16Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.