setup.py

try:
    from setuptools import setup
except ImportError:
    from distutils.core import setup

setup(
    name='classification_models_3D',
    version='1.1.0',
    author='Roman Sol (ZFTurbo)',
    packages=['classification_models_3D', 'classification_models_3D/models'],
    url='https://github.com/ZFTurbo/classification_models_3D',
    description='Set of models for classification of 3D volumes.',
    long_description='This repository contains 3D variants of popular classification CNN models like ResNets, DenseNets, VGG, etc for keras module. '
                     'It also contains weights obtained by converting ImageNet weights from the same 2D models. '
                     'Models work with keras 3 with different backends like Tensorflow, Torch and Jax.'
                     'More details: https://github.com/ZFTurbo/classification_models_3D',
)