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)