def list_weights(
root=config["darknet"]["root"], weights=config["darknet"]["weight_dir"]
):
"""List weights available in weights directory."""
darknet = pydarknet2.Darknet(root=root, weight_dir=weights)
for weight in darknet.weights:
print(weight)
def test_clone_000003_04(darknet_root, clone_url):
# Build with opencv or openmp.
dn = pydarknet2.Darknet(root=darknet_root)
dn.build(opencv=True, openmp=True, force=True)
out = subprocess.check_output(["ldd", dn.exe]).decode("UTF-8")
assert "openmp" in out
assert "opencv" in out
def clone(
root=config["darknet"]["root"], url=config["darknet"]["clone_url"], force=False,
):
"""Clone a darknet repository."""
click.echo("URL: %s" % url)
click.echo("Root: %s" % root)
darknet = pydarknet2.Darknet(root=root)
darknet.clone(clone_url=url, force=force)
def test_clone_000003_05(darknet_root, clone_url):
# Build with GPU & cudNN if available.
dn = pydarknet2.Darknet(root=darknet_root)
# Hack/Workaround.
os.environ[
"PATH"] = os.environ["PATH"] + os.pathsep + "/usr/local/cuda/bin"
if "LD_LIBRARY_PATH" in os.environ:
os.environ["LD_LIBRARY_PATH"] = (os.environ["LD_LIBRARY_PATH"] +
os.pathsep + "/usr/local/cuda/lib64")
else:
os.environ["LD_LIBRARY_PATH"] = "/usr/local/cuda/lib64"
dn.build(gpu=True, cudnn=True, force=True)
out = subprocess.check_output(["ldd", dn.exe]).decode("UTF-8")
assert "libcudart" in out
assert "libcudnn" in out
def build(gpu, cudnn, opencv, openmp, force, root):
"""Build darknet."""
darknet = pydarknet2.Darknet(root=root)
darknet.build(gpu=gpu, cudnn=cudnn, opencv=opencv, openmp=openmp, force=force)
def test_build_000003_01(darknet_root, clone_url):
# Clone a darknet directory.
dn = pydarknet2.Darknet(root=darknet_root)
dn.clone(clone_url=clone_url)