apolinario's picture
upload clipseg
48fa639
raw
history blame
787 Bytes
from setuptools import setup
with open("README.md", "r", encoding="utf-8") as readme_file:
readme = readme_file.read()
requirements = [
"numpy",
"scipy",
"matplotlib",
"torch",
"torchvision",
"opencv-python",
"CLIP @ git+https://github.com/openai/CLIP.git"
]
setup(
name='clipseg',
packages=['clipseg'],
package_dir={'clipseg': 'models'},
package_data={'clipseg': [
"../weights/*.pth",
]},
version='0.0.1',
url='https://github.com/timojl/clipseg',
python_requires='>=3.9',
install_requires=requirements,
description='This repository contains the code used in the paper "Image Segmentation Using Text and Image Prompts".',
long_description=readme,
long_description_content_type="text/markdown",
)