diff --git a/README.md b/README.md index e3281da..e042f33 100644 --- a/README.md +++ b/README.md @@ -38,7 +38,9 @@ In order to run this toolbox you will need: ### Usage -Navigate (```cd```) to the root of the toolbox ```[YOUR_CIRTORCH_ROOT]```. +Navigate (```cd```) to the root of the toolbox ```[YOUR_CIRTORCH_ROOT]```. +You can install package with `pip3 install .` if you need. +Make sure to have desired PyTorch and torchvision packages installed.
Training
diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..6c93b92 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,2 @@ +torch==1.0.0 +torchvision==0.2.1 diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..e5d0548 --- /dev/null +++ b/setup.py @@ -0,0 +1,32 @@ +import os + +from setuptools import find_packages, setup + +with open(os.path.join(os.path.dirname(__file__), "README.md"), encoding="utf-8") as f: + long_description = f.read() + +setup( + name="cirtorch", + description="CNN Image Retrieval in PyTorch: " + "Training and evaluating CNNs for Image Retrieval in PyTorch", + long_description=long_description, + long_description_content_type="text/markdown", + version="0.0.1", + url="https://github.com/filipradenovic/cnnimageretrieval-pytorch", + download_url="https://github.com/filipradenovic/cnnimageretrieval-pytorch", + packages=find_packages(), + keywords=["machine learning", "cnn", "computer vision"], + install_requires=[ + "torch>=1.0.0,<1.4.0", + "torchvision", + ], + classifiers=[ + "Development Status :: 3 - Alpha", + "Environment :: Console", + "Intended Audience :: Developers", + "Intended Audience :: Science/Research", + "Operating System :: POSIX", + "Programming Language :: Python :: 3.7", + "License :: OSI Approved :: MIT License", + ], +)