Mercurial Hosting > traffic-intelligence
comparison trafficintelligence/storage.py @ 1049:c9c03c97ed9f
bug fix to store numpy integers in SQLite
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 06 Jul 2018 15:58:45 -0400 |
parents | 75a6ad604cc5 |
children | 9d4a06f49cb8 d13f9bfbf3ff |
comparison
equal
deleted
inserted
replaced
1048:27a822922cb0 | 1049:c9c03c97ed9f |
---|---|
5 from pathlib import Path | 5 from pathlib import Path |
6 import shutil | 6 import shutil |
7 from copy import copy | 7 from copy import copy |
8 import sqlite3, logging | 8 import sqlite3, logging |
9 | 9 |
10 from numpy import log, min as npmin, max as npmax, round as npround, array, sum as npsum, loadtxt, floor as npfloor, ceil as npceil, linalg | 10 from numpy import log, min as npmin, max as npmax, round as npround, array, sum as npsum, loadtxt, floor as npfloor, ceil as npceil, linalg, int32, int64 |
11 from pandas import read_csv, merge | 11 from pandas import read_csv, merge |
12 | 12 |
13 from trafficintelligence import utils, moving, events, indicators | 13 from trafficintelligence import utils, moving, events, indicators |
14 from trafficintelligence.base import VideoFilenameAddable | 14 from trafficintelligence.base import VideoFilenameAddable |
15 | 15 |
19 'truck':3} | 19 'truck':3} |
20 | 20 |
21 tableNames = {'feature':'positions', | 21 tableNames = {'feature':'positions', |
22 'object': 'objects', | 22 'object': 'objects', |
23 'objectfeatures': 'positions'} | 23 'objectfeatures': 'positions'} |
24 | |
25 sqlite3.register_adapter(int64, lambda val: int(val)) | |
26 sqlite3.register_adapter(int32, lambda val: int(val)) | |
24 | 27 |
25 ######################### | 28 ######################### |
26 # Sqlite | 29 # Sqlite |
27 ######################### | 30 ######################### |
28 | 31 |