In [1]:

    

import sys, os
sys.path.append(os.path.join(os.getcwd(),'python/'))

from PIL import Image, ImageFont, ImageDraw, ImageEnhance


import darknet as dn
import pdb

dn.set_gpu(0)

net = dn.load_net("cfg/yolov3.cfg", "yolov3.weights", 0)
meta = dn.load_meta("cfg/coco.data")


def detect_im(in_path, out_path):
    im = Image.open(in_path)
    im = im.convert('RGBA')
    tmp = Image.new('RGBA', im.size, (0,0,0,0))
    r = dn.detect(net, meta, in_path)
    draw = ImageDraw.Draw(tmp)
    for cat, score, rect in r:
        x, y, w, h = rect
        draw.rectangle(((x-w/2, y-h/2), (x+w/2, y-h/2+20)), fill=(0,0,0,135), outline=None)
        draw.rectangle(((x-w/2, y-h/2), (x+w/2, y+h/2)), fill=None, outline="red")
        draw.text((x-w/2+4, y-h/2), cat, font=ImageFont.truetype("Arial.ttf", 16))
    img = Image.alpha_composite(im, tmp)
    img = img.convert("RGB") 
    img.save(out_path, 'PNG')

    
in_dir = 'kanye/frames/'
out_dir = 'kanye/frames2/'
for i in range(0,3500):#3500
    if i % 100 == 0:
        print(i)
    f_in = '%s/f%05d.png'%(in_dir,i)
    f_out = '%s/f%05d.png'%(out_dir,i)
    detect_im(f_in, f_out)


    

    




---------------------------------------------------------------------------
OSError                                   Traceback (most recent call last)
<ipython-input-1-d7ad36311053> in <module>()
     11 
     12 
---> 13 import darknet as dn
     14 import pdb
     15 

/mnt/drive2/gene/ml4a-guides/tools/darknet/python/darknet.py in <module>()
     45 
     46 
---> 47 lib = CDLL("/mnt/drive2/gene/ml4a-guides/tools/darknet/libdarknet.so", RTLD_GLOBAL)
     48 #lib = CDLL("libdarknet.so", RTLD_GLOBAL)
     49 lib.network_width.argtypes = [c_void_p]

/usr/lib/python2.7/ctypes/__init__.pyc in __init__(self, name, mode, handle, use_errno, use_last_error)
    360 
    361         if handle is None:
--> 362             self._handle = _dlopen(self._name, mode)
    363         else:
    364             self._handle = handle

OSError: /lib/x86_64-linux-gnu/libz.so.1: version `ZLIB_1.2.9' not found (required by /mnt/drive1/tools/anaconda2/lib/libpng16.so.16)

In [2]:

    

import darknet as dn


    

    




---------------------------------------------------------------------------
OSError                                   Traceback (most recent call last)
<ipython-input-2-f144c6b7ec2f> in <module>()
----> 1 import darknet as dn

/mnt/drive2/gene/ml4a-guides/tools/darknet/python/darknet.py in <module>()
     45 
     46 
---> 47 lib = CDLL("/mnt/drive2/gene/ml4a-guides/tools/darknet/libdarknet.so", RTLD_GLOBAL)
     48 #lib = CDLL("libdarknet.so", RTLD_GLOBAL)
     49 lib.network_width.argtypes = [c_void_p]

/usr/lib/python2.7/ctypes/__init__.pyc in __init__(self, name, mode, handle, use_errno, use_last_error)
    360 
    361         if handle is None:
--> 362             self._handle = _dlopen(self._name, mode)
    363         else:
    364             self._handle = handle

OSError: /lib/x86_64-linux-gnu/libz.so.1: version `ZLIB_1.2.9' not found (required by /mnt/drive1/tools/anaconda2/lib/libpng16.so.16)

In [3]:

    

from darknet import *


    

    




Traceback (most recent call last):

  File "/home/gene/.local/lib/python3.5/site-packages/IPython/core/interactiveshell.py", line 2910, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)

  File "<ipython-input-3-1e856bc178f2>", line 1, in <module>
    from darknet import *

  File "../tools/darknet/python/darknet.py", line 154
    print r
          ^
SyntaxError: Missing parentheses in call to 'print'

In [ ]: