Mercurial Hosting > traffic-intelligence
changeset 867:003445db1e30
forgot annotation mask filename
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 08 Dec 2016 18:00:53 -0500 |
parents | 8fba46899e74 |
children | 1fdafa9f6bf4 |
files | python/metadata.py |
diffstat | 1 files changed, 9 insertions(+), 1 deletions(-) [+] |
line wrap: on
line diff
--- a/python/metadata.py Thu Dec 08 17:51:03 2016 -0500 +++ b/python/metadata.py Thu Dec 08 18:00:53 2016 -0500 @@ -314,17 +314,19 @@ firstFrameNum = Column(Integer) # first frame num of annotated data (could be computed on less data) lastFrameNum = Column(Integer) videoSequenceIdx = Column(Integer, ForeignKey('video_sequences.idx')) + maskFilename = Column(String) # path to mask file (can be different from camera view, for annotations), relative to site name undistorted = Column(Boolean) # indicates whether the annotations were done in undistorted video space videoSequence = relationship("VideoSequence", backref=backref('video_sequences', order_by = idx)) - def __init__(self, description, groundTruthFilename, firstFrameNum, lastFrameNum, videoSequence, undistorted = True): + def __init__(self, description, groundTruthFilename, firstFrameNum, lastFrameNum, videoSequence, maskFilename, undistorted = True): self.description = description self.groundTruthFilename = groundTruthFilename self.firstFrameNum = firstFrameNum self.lastFrameNum = lastFrameNum self.videoSequence = videoSequence self.undistorted = undistorted + self.maskFilename = maskFilename def getGroundTruthFilename(self, relativeToSiteFilename = True): if relativeToSiteFilename: @@ -332,6 +334,12 @@ else: return self.groundTruthFilename + def getMaskFilename(self, relativeToSiteFilename = True): + if relativeToSiteFilename: + return path.join(self.videoSequence.cameraView.site.getPath(), self.maskFilename) + else: + return self.maskFilename + def getTimeInterval(self): return TimeInterval(self.firstFrameNum, self.lastFrameNum)