Merge pull request #56 from zurk/setup

Add setup.py

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.  
 
 <details>
   <summary><b>Training</b></summary><br/>

requirements.txt

@@ -0,0 +1,2 @@
+torch==1.0.0
+torchvision==0.2.1

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",
+    ],
+)