• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• refinedet.pytorch-master
  • demo
    • demo_simple.py
  • _init_paths.py
  • libs
    • datasets
      • pascal_voc_rbg.py
      • coco.py
      • imdb.py
      • voc_eval.py
      • factory.py
      • vg.py
      • __init__.py
      • vg_eval.py
      • pascal_voc.py
      • imagenet.py
      • ds_utils.py
    • utils
      • net_utils.py
      • path_config.py
      • config.py
      • __init__.py
      • box_utils.py
      • augmentations.py
    • networks
      • vgg.py
      • vgg_refinedet.py
      • refinedet.py
      • __init__.py
      • resnet_refinedet.py
      • resnet101.py
    • setup.py
    • pycocotools
      • license.txt
      • coco.py
      • UPSTREAM_REV
      • maskApi.h
      • mask.py
      • cocoeval.py
      • maskApi.c
      • __init__.py
      • _mask.pyx
    • __init__.py
    • data_layers
      • blob_dataset.py
      • transform.py
      • roidb.py
      • __init__.py
    • modules
      • arm_loss.py
      • odm_loss.py
      • detect_layer.py
      • __init__.py
      • anchor_box.py
    • make.sh
  • start_train_resnet101_coco512.sh
  • train_refinedet.py
  • eval_refinedet.py
  • start_train_vgg16_voc320.sh
  • LICENSE
  • __init__.py
  • README.md
  • kill_train.sh
  • start_train_vgg16_voc512.sh
  • requirements.txt
  • .gitignore
  • eval.sh
  • start_train_resnet101_coco320.sh

from __future__ import print_function
# --------------------------------------------------------
# Fast R-CNN
# Copyright (c) 2015 Microsoft
# Licensed under The MIT License [see LICENSE for details]
# Written by Ross Girshick
# --------------------------------------------------------

import os
from os.path import join as pjoin
import numpy as np
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext


def find_in_path(name, path):
    "Find a file in a search path"
    # adapted fom http://code.activestate.com/recipes/52224-find-a-file-given-a-search-path/
    for dir in path.split(os.pathsep):
        binpath = pjoin(dir, name)
        if os.path.exists(binpath):
            return os.path.abspath(binpath)
    return None



# Obtain the numpy include directory.  This logic works across numpy versions.
try:
    numpy_include = np.get_include()
except AttributeError:
    numpy_include = np.get_numpy_include()


def customize_compiler_for_nvcc(self):
    """inject deep into distutils to customize how the dispatch
    to gcc/nvcc works.

    If you subclass UnixCCompiler, it's not trivial to get your subclass
    injected in, and still have the right customizations (i.e.
    distutils.sysconfig.customize_compiler) run on it. So instead of going
    the OO route, I have this. Note, it's kindof like a wierd functional
    subclassing going on."""

    # tell the compiler it can processes .cu
    self.src_extensions.append('.cu')

    # save references to the default compiler_so and _comple methods
    default_compiler_so = self.compiler_so
    super = self._compile

    # now redefine the _compile method. This gets executed for each
    # object but distutils doesn't have the ability to change compilers
    # based on source extension: we add it.
    def _compile(obj, src, ext, cc_args, extra_postargs, pp_opts):
        print(extra_postargs)
        if os.path.splitext(src)[1] == '.cu':
            # use the cuda for .cu files
            self.set_executable('compiler_so', CUDA['nvcc'])
            # use only a subset of the extra_postargs, which are 1-1 translated
            # from the extra_compile_args in the Extension class
            postargs = extra_postargs['nvcc']
        else:
            postargs = extra_postargs['gcc']

        super(obj, src, ext, cc_args, postargs, pp_opts)
        # reset the default compiler_so, which we might have changed for cuda
        self.compiler_so = default_compiler_so

    # inject our redefined _compile method into the class
    self._compile = _compile


# run the customize_compiler
class custom_build_ext(build_ext):
    def build_extensions(self):
        customize_compiler_for_nvcc(self.compiler)
        build_ext.build_extensions(self)


ext_modules = [
    Extension(
        'pycocotools._mask',
        sources=['pycocotools/maskApi.c', 'pycocotools/_mask.pyx'],
        include_dirs=[numpy_include, 'pycocotools'],
        extra_compile_args={
            'gcc': ['-Wno-cpp', '-Wno-unused-function', '-std=c99']},
    ),
]

setup(
    name='refinedet_pytorch',
    ext_modules=ext_modules,
    # inject our custom trigger
    cmdclass={'build_ext': custom_build_ext},
)