Mercurial > hg > nsaunier > traffic-intelligence

comparison trafficintelligence/storage.py @ 1049:c9c03c97ed9f

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