def run():
detector = PeopleDetector()
frame = cv2.imread("9.png")
detected_points = detector.detect(frame, debug=True, degree=10)
# Extend to rectangle
for p in detected_points:
p.append(10)
p.append(10)
PeopleDetector.draw_detections(frame, detected_points, 3)
cv2.imshow("test", frame)
cv2.waitKey(0)
class PeopleEvidence:
def __init__(self):
# People detector
self.people_detector = PeopleDetector()
def get_evidence(self, frame):
people_squares = self.people_detector.detect(frame, degree=30)
rows, cols = frame.shape[0], frame.shape[1]
boundary = [[0, 0], [rows, 0], [rows, cols], [0, rows]]
evidence = Evidence(boundary, PEOPLE_SENSOR, BOOLEAN_STATES, BOOLEAN_STATES[0])
# Square to evidence
for square in people_squares:
evidence.add_detection(BOOLEAN_STATES[1], square)
return evidence
def __init__(self):
# People detector
self.people_detector = PeopleDetector()
from people_detector import PeopleDetector
from utils import user_input, data
if __name__ == "__main__":
args = user_input.parse_arguments()
if not args.filenames:
filenames = data.get_filenames(args.data_dir)
else:
filenames = [f.strip() for f in args.filenames.split(",")]
people_detector = PeopleDetector(args.yolo_config_path)
people_detector.detect(filenames, args.data_dir, args.results_dir)
