Skip to content

Commit 4fca9df

Browse files
committed
update setup.py
1 parent cb23651 commit 4fca9df

File tree

2 files changed

+14
-27
lines changed

2 files changed

+14
-27
lines changed

main.py

-24
Original file line numberDiff line numberDiff line change
@@ -7,36 +7,12 @@
77
Please see the LICENSE file that should have been included as part of this package.
88
"""
99

10-
import os
11-
import cv2
12-
import matplotlib.pyplot as plt
13-
import gdown
14-
import zipfile36 as zipfile
15-
16-
1710
from ColorTransferLib.MeshProcessing.Mesh import Mesh
1811
from ColorTransferLib.ImageProcessing.Image import Image
1912
from ColorTransferLib.ColorTransfer import ColorTransfer, ColorTransferEvaluation
2013

2114

2215

23-
# ------------------------------------------------------------------------------------------------------------------
24-
# Downloads necessary objects, e.g., network models, to the correct location
25-
# ------------------------------------------------------------------------------------------------------------------
26-
def downloadMaterial():
27-
if not os.path.exists("Models") and not os.path.exists("data"):
28-
print("Download DATA.zip ...")
29-
url = "https://drive.google.com/file/d/1ShJpPFJ9vCu5Vb7FJk7aSnn7ywFiT0GJ/view?usp=share_link"
30-
output_path = 'DATA.zip'
31-
gdown.download(url, output_path, quiet=False, fuzzy=True)
32-
# Extract DATA.zip
33-
print("Extract DATA.zip ...")
34-
with zipfile.ZipFile("DATA.zip","r") as zip_ref:
35-
zip_ref.extractall()
36-
# Delete DATA.zip
37-
print("Delete DATA.zip ...")
38-
os.remove("DATA.zip")
39-
4016
# ------------------------------------------------------------------------------------------------------------------
4117
# ------------------------------------------------------------------------------------------------------------------
4218
#

setup.py

+14-3
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,24 @@
11
import setuptools
22

3+
from pathlib import Path
4+
this_directory = Path(__file__).parent
5+
long_description = (this_directory / "README.md").read_text()
6+
7+
# Reading the content of the requirements.txt
8+
with open('requirements/requirements.txt') as f:
9+
requirements = f.read().splitlines()
10+
311
setuptools.setup(
412
name="ColorTransferLib",
5-
version="0.0.2",
13+
version="1.0.0-2",
614
author="Herbert Potechius",
715
author_email="[email protected]",
8-
description="This library provides color transfer algorithms which were published in scientific papers.",
16+
description="This library provides color and tyle transfer algorithms which were published in scientific papers. Additionall a set of IQA metrics are available.",
917
packages=setuptools.find_packages(),
1018
classifiers=[
1119
"Programming Language :: Python :: 3",
1220
"License :: OSI Approved :: MIT License",
13-
"Operating System :: OS Independent",
21+
"Operating System :: POSIX :: Linux",
1422
],
1523
package_data={"": ['Options/*.json',
1624
'Config/*.json',
@@ -22,4 +30,7 @@
2230
'Evaluation/VIS/gbvs/*/*/*']},
2331
include_package_data=True,
2432
python_requires='>=3.8',
33+
long_description=long_description,
34+
long_description_content_type='text/markdown',
35+
install_requires=requirements,
2536
)

0 commit comments

Comments
 (0)