Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
matterport
GitHub Repository: matterport/Mask_RCNN
 Path: blob/master/setup.py
239 views
1
"""

2
The build/compilations setup

3


4
>> pip install -r requirements.txt

5
>> python setup.py install

6
"""

7
import pip

8
import logging

9
import pkg_resources

10
try:

11
    from setuptools import setup

12
except ImportError:

13
    from distutils.core import setup

14


15


16
def _parse_requirements(file_path):

17
    pip_ver = pkg_resources.get_distribution('pip').version

18
    pip_version = list(map(int, pip_ver.split('.')[:2]))

19
    if pip_version >= [6, 0]:

20
        raw = pip.req.parse_requirements(file_path,

21
                                         session=pip.download.PipSession())

22
    else:

23
        raw = pip.req.parse_requirements(file_path)

24
    return [str(i.req) for i in raw]

25


26


27
# parse_requirements() returns generator of pip.req.InstallRequirement objects

28
try:

29
    install_reqs = _parse_requirements("requirements.txt")

30
except Exception:

31
    logging.warning('Fail load requirements file, so using default ones.')

32
    install_reqs = []

33


34
setup(

35
    name='mask-rcnn',

36
    version='2.1',

37
    url='https://github.com/matterport/Mask_RCNN',

38
    author='Matterport',

39
    author_email='[email protected]',

40
    license='MIT',

41
    description='Mask R-CNN for object detection and instance segmentation',

42
    packages=["mrcnn"],

43
    install_requires=install_reqs,

44
    include_package_data=True,

45
    python_requires='>=3.4',

46
    long_description="""This is an implementation of Mask R-CNN on Python 3, Keras, and TensorFlow. 

47
The model generates bounding boxes and segmentation masks for each instance of an object in the image. 

48
It's based on Feature Pyramid Network (FPN) and a ResNet101 backbone.""",

49
    classifiers=[

50
        "Development Status :: 5 - Production/Stable",

51
        "Environment :: Console",

52
        "Intended Audience :: Developers",

53
        "Intended Audience :: Information Technology",

54
        "Intended Audience :: Education",

55
        "Intended Audience :: Science/Research",

56
        "License :: OSI Approved :: MIT License",

57
        "Natural Language :: English",

58
        "Operating System :: OS Independent",

59
        "Topic :: Scientific/Engineering :: Artificial Intelligence",

60
        "Topic :: Scientific/Engineering :: Image Recognition",

61
        "Topic :: Scientific/Engineering :: Visualization",

62
        "Topic :: Scientific/Engineering :: Image Segmentation",

63
        'Programming Language :: Python :: 3.4',

64
        'Programming Language :: Python :: 3.5',

65
        'Programming Language :: Python :: 3.6',

66
    ],

67
    keywords="image instance segmentation object detection mask rcnn r-cnn tensorflow keras",

68
)

69


70