#?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個文件信息