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)