dlib-obj-counting-init.py

import os
import sys
import glob

import dlib
from skimage import io
from skimage.draw import polygon_perimeter
import traceback

dlib-obj-counting-load.py

detector = dlib.simple_object_detector(detector_svm)

dlib-obj-counting-option-c.py

options.C = 5

dlib-obj-counting-option-cpu.py

options.num_threads = 4
options.be_verbose = True

dlib-obj-counting-optioninit.py

options = dlib.simple_object_detector_training_options()

dlib-obj-counting-optionsym.py

options.add_left_right_image_flips = True

dlib-obj-counting-run.py

print("Showing detections on the images in the test folder...")
win = dlib.image_window()
for f in glob.glob(os.path.join(detector_folder, "test/*.jpg")):
    print("Processing file: {}".format(f))
    img = io.imread(f)
    dets = detector(img)
    print("Number of objects detected: {}".format(len(dets)))
    bOverLays = False
    for k, d in enumerate(dets):
        print("Detection {}: Left: {} Top: {} Right: {} Bottom: {}".format(
            k, d.left(), d.top(), d.right(), d.bottom()))
        rr,cc = polygon_perimeter([d.top(), d.top(), d.bottom(), d.bottom()],
                             [d.right(), d.left(), d.left(), d.right()])
        try:
            img[rr, cc] = (255, 0, 0)
            if bOverLays == False:
                bOverLays = True
        except:
            traceback.print_exc()
    # Save the image detections to a file for future review.
    if bOverLays == True:
        io.imsave(f.replace("test/","output/"), img)
    win.clear_overlay()
    win.set_image(img)
    win.add_overlay(dets)
    dlib.hit_enter_to_continue()

dlib-obj-counting-test-accuracy.py

print("Testing accuracy: {}".format(
    dlib.test_simple_object_detector(testing_xml_path, detector_svm)))

dlib-obj-counting-test.py

print("")  # Print blank line to create gap from previous output
print("Training accuracy: {}".format(
    dlib.test_simple_object_detector(training_xml_path, detector_svm)))

dlib-obj-counting-train.py

if not os.path.exists(detector_svm):
    dlib.train_simple_object_detector(training_xml_path, detector_svm, options)