Python-用PyTorch10实现FasterRCNN和MaskRCNN比Detectron快2倍

Facebook发布，基于PyTorch 1.0的高性能图像检测/分割实现，比Detectron快2倍

# Copyright （c） Facebook Inc. and its affiliates. All Rights Reserved.
#!/usr/bin/env python

import glob
import os

import torch
from setuptools import find_packages
from setuptools import setup
from torch.utils.cpp_extension import CUDA_HOME
from torch.utils.cpp_extension import CppExtension
from torch.utils.cpp_extension import CUDAExtension

requirements = [“torch“ “torchvision“]


def get_extensions（）:
    this_dir = os.path.dirname（os.path.abspath（__file__））
    extensions_dir = os.path.join（this_dir “maskrcnn_benchmark“ “csrc“）

    main_file = glob.glob（os.path.join（extensions_dir “*.cpp“））
    source_cpu = glob.glob（os.path.join（extensions_dir “cpu“ “*.cpp“））
    source_cuda = glob.glob（os.path.join（extensions_dir “cuda“ “*.cu“））

    sources = main_file + source_cpu
    extension = CppExtension

    extra_compile_args = {“cxx“: []}
    define_macros = []

    if （torch.cuda.is_available（） and CUDA_HOME is not None） or os.getenv（“FORCE_CUDA“ “0“） == “1“:
        extension = CUDAExtension
        sources += source_cuda
        define_macros += [（“WITH_CUDA“ None）]
        extra_compile_args[“nvcc“] = [
            “-DCUDA_HAS_FP16=1“
            “-D__CUDA_NO_HALF_OPERATORS__“
            “-D__CUDA_NO_HALF_CONVERSIONS__“
            “-D__CUDA_NO_HALF2_OPERATORS__“
        ]

    sources = [os.path.join（extensions_dir s） for s in sources]

    include_dirs = [extensions_dir]

    ext_modules = [
        extension（
            “maskrcnn_benchmark._C“
            sources
            include_dirs=include_dirs
            define_macros=define_macros
            extra_compile_args=extra_compile_args
        ）
    ]

    return ext_modules


setup（
    name=“maskrcnn_benchmark“
    version=“0.1“
    author=“fmassa“
    url=“https://github.com/facebookresearch/maskrcnn-benchmark“
    description=“object detection in pytorch“
    packages=find_packages（exclude=（“configs“ “tests“））
    # install_requires=requirements
    ext_modules=get_extensions（）
    cmdclass={“build_ext“: torch.utils.cpp_extension.BuildExtension}
）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2019-07-12 09:11  maskrcnn-benchmark-master\
     文件         247  2019-07-12 09:11  maskrcnn-benchmark-master\.flake8
     目录           0  2019-07-12 09:11  maskrcnn-benchmark-master\.github\
     目录           0  2019-07-12 09:11  maskrcnn-benchmark-master\.github\ISSUE_TEMPLATE\
     文件        1269  2019-07-12 09:11  maskrcnn-benchmark-master\.github\ISSUE_TEMPLATE\bug-report.md
     文件         730  2019-07-12 09:11  maskrcnn-benchmark-master\.github\ISSUE_TEMPLATE\feature-request.md
     文件          98  2019-07-12 09:11  maskrcnn-benchmark-master\.github\ISSUE_TEMPLATE\questions-help-support.md
     文件         382  2019-07-12 09:11  maskrcnn-benchmark-master\.gitignore
     文件        2654  2019-07-12 09:11  maskrcnn-benchmark-master\ABSTRACTIONS.md
     文件         244  2019-07-12 09:11  maskrcnn-benchmark-master\CODE_OF_CONDUCT.md
     文件        1641  2019-07-12 09:11  maskrcnn-benchmark-master\CONTRIBUTING.md
     文件        4182  2019-07-12 09:11  maskrcnn-benchmark-master\INSTALL.md
     文件        1065  2019-07-12 09:11  maskrcnn-benchmark-master\LICENSE
     文件        6877  2019-07-12 09:11  maskrcnn-benchmark-master\MODEL_ZOO.md
     文件       13616  2019-07-12 09:11  maskrcnn-benchmark-master\README.md
     文件        2938  2019-07-12 09:11  maskrcnn-benchmark-master\TROUBLESHOOTING.md
     目录           0  2019-07-12 09:11  maskrcnn-benchmark-master\configs\
     目录           0  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\
     文件         701  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_faster_rcnn_R_101_FPN_1x_caffe2.yaml
     文件         166  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_faster_rcnn_R_50_C4_1x_caffe2.yaml
     文件         699  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_faster_rcnn_R_50_FPN_1x_caffe2.yaml
     文件         774  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_faster_rcnn_X_101_32x8d_FPN_1x_caffe2.yaml
     文件        1251  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_keypoint_rcnn_R_50_FPN_1x_caffe2.yaml
     文件         985  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_mask_rcnn_R_101_FPN_1x_caffe2.yaml
     文件         272  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_mask_rcnn_R_50_C4_1x_caffe2.yaml
     文件         983  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_mask_rcnn_R_50_FPN_1x_caffe2.yaml
     文件        1066  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_mask_rcnn_X-152-32x8d-FPN-IN5k_1.44x_caffe2.yaml
     文件        1058  2019-07-12 09:11  maskrcnn-benchmark-master\configs\caffe2\e2e_mask_rcnn_X_101_32x8d_FPN_1x_caffe2.yaml
     目录           0  2019-07-12 09:11  maskrcnn-benchmark-master\configs\cityscapes\
     文件        6171  2019-07-12 09:11  maskrcnn-benchmark-master\configs\cityscapes\README.md
     文件         864  2019-07-12 09:11  maskrcnn-benchmark-master\configs\cityscapes\e2e_faster_rcnn_R_50_FPN_1x_cocostyle.yaml
............此处省略262个文件信息