benchmark_darknet_api.py

import random
import requests
import time


#url = 'http://tgpu102.imgr.nhnsystem.com:30001/darknet1.search?worker_id=1'
url = 'http://tgpu102.imgr.nhnsystem.com:41901'

IMAGE = 'dog{}.jpg'

N = 500
s = time.time()
response_times = []
for i in range(N):

  filename = 'img/' + IMAGE.format(random.randint(1, 5))
  with open(filename, 'rb') as f:
    r_start = time.time()
    r = requests.post(url, files={'image': f})
    r_duration = time.time() - r_start
    response_times.append(r_duration)
    if not r.ok:
      print(i, r)

d = time.time()-s

print()
print('url: {}'.format(url))
print('-'*30)
print('{} seconds'.format(d))
print('{} qps'.format(N/d))

print('response time(avg): {} ms'.format(sum(response_times)/len(response_times)*1000))