def run(self): tempdir = tempfile.gettempdir() darknet_url = 'https://github.com/pjreddie/darknet/archive/61c9d02ec461e30d55762ec7669d6a1d3c356fb2.zip' darknet_zip_file = os.path.join(tempdir, 'darknet.zip') darknet_root = os.path.join( tempdir, 'darknet-61c9d02ec461e30d55762ec7669d6a1d3c356fb2') makefile = os.path.join(darknet_root, 'Makefile') os.environ['DARKNET_ROOT'] = darknet_root retriver.urlretrieve(darknet_url, darknet_zip_file) with zipfile.ZipFile(darknet_zip_file, 'r') as zip_ref: zip_ref.extractall(tempdir) if (os.environ.get('GPU', None)): sed('GPU=0', 'GPU=1', makefile, count=1) if (os.environ.get('CUDNN', None)): sed('CUDNN=0', 'CUDNN=1', makefile, count=1) if (os.environ.get('OPENCV', None)): sed('OPENCV=0', 'OPENCV=1', makefile, count=1) if (os.environ.get('OPENMP', None)): sed('OPENMP=0', 'OPENMP=1', makefile, count=1) if (os.environ.get('DEBUG', None)): sed('DEBUG=0', 'DEBUG=1', makefile, count=1) process = subprocess.Popen('make', cwd=darknet_root, shell=True) process.wait() self.include_dirs.append(tempdir) self.include_dirs.append(os.path.join(darknet_root, 'include')) self.library_dirs.append(darknet_root) self.libraries.append('darknet') build_ext.run(self)
def run(self): build_ext.run(self)