|
|
|
|
|
from setuptools import setup, find_packages |
|
|
|
|
|
setup( |
|
|
name="SemanticModel", |
|
|
version="0.1.0", |
|
|
description="Deep learning framework for semantic segmentation", |
|
|
author="Your Name", |
|
|
packages=find_packages(), |
|
|
python_requires=">=3.8", |
|
|
install_requires=[ |
|
|
'torch', |
|
|
'torchvision', |
|
|
'tensorboard', |
|
|
'pyproj', |
|
|
'fiona==1.8.20', |
|
|
'rtree', |
|
|
'geopandas', |
|
|
'rasterio', |
|
|
'slidingwindow', |
|
|
'opencv-python', |
|
|
'wandb', |
|
|
'tifffile', |
|
|
'imagecodecs', |
|
|
'albumentations', |
|
|
'segmentation-models-pytorch>=0.3.3' |
|
|
], |
|
|
classifiers=[ |
|
|
"Development Status :: 3 - Alpha", |
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence", |
|
|
"License :: OSI Approved :: MIT License", |
|
|
"Programming Language :: Python :: 3.8", |
|
|
], |
|
|
) |
