Mercurial Hosting > traffic-intelligence
comparison scripts/clean-ground-truth.py @ 782:08f82be22816 dev
added script to detect issues in ground truth databases
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 25 Feb 2016 17:14:22 -0500 |
parents | |
children | 933670761a57 |
comparison
equal
deleted
inserted
replaced
781:7c38250ddfc7 | 782:08f82be22816 |
---|---|
1 #! /usr/bin/env python | |
2 import argparse | |
3 import pandas as pd | |
4 import sqlite3 | |
5 from numpy import min, max | |
6 | |
7 #import utils | |
8 #import storage | |
9 | |
10 parser = argparse.ArgumentParser(description='The program finds ground truth annotation objects with missing positions (and the times that do not have a position') | |
11 #parser.add_argument('configFilename', help = 'name of the configuration file') | |
12 parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the Ground Truth Sqlite database', required = True) | |
13 | |
14 args = parser.parse_args() | |
15 | |
16 data = pd.read_sql_query("select object_id, frame_number from bounding_boxes", sqlite3.connect(args.databaseFilename)) | |
17 #aggData = data.groupby(["object_id"]).agg([min, max, len]) | |
18 | |
19 nProblems = 0 | |
20 for object_id in data["object_id"].unique(): | |
21 tmp = data[data["object_id"] == object_id] | |
22 firstInstant = min(tmp["frame_number"]) | |
23 lastInstant = max(tmp["frame_number"]) | |
24 if lastInstant-firstInstant != len(tmp)-1: | |
25 nProblems += 1 | |
26 times=set(range(firstInstant, lastInstant+1)) | |
27 print("Object {} has missing positions at instants: {}".format(object_id, times.difference(tmp["frame_number"]))) | |
28 | |
29 if nProblems == 0: | |
30 print("No problems were detected in database {}".format(args.databaseFilename)) |