Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/clean-ground-truth.py @ 782:08f82be22816 dev

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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))