fixed dump of interpolation points object && statistics calculation (#…

…1108)
cvat-ai · Jan 30, 2020 · e062c23 · e062c23
1 parent b918aeb
commit e062c23
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/cvat/apps/engine/data_manager.py b/cvat/apps/engine/data_manager.py
@@ -357,7 +357,8 @@ def interpolate(shape0, shape1):
             prev_shape = shape
 
         # TODO: Need to modify a client and a database (append "outside" shapes for polytracks)
-        if not prev_shape["outside"] and prev_shape["type"] == models.ShapeType.RECTANGLE:
+        if not prev_shape["outside"] and (prev_shape["type"] == models.ShapeType.RECTANGLE
+               or prev_shape["type"] == models.ShapeType.POINTS):
             shape = copy.copy(prev_shape)
             shape["frame"] = end_frame
             shapes.extend(interpolate(prev_shape, shape))

diff --git a/cvat/apps/engine/static/engine/js/shapes.js b/cvat/apps/engine/static/engine/js/shapes.js
@@ -197,7 +197,8 @@ class ShapeModel extends Listener {
         }
 
         if (visibleFrame != null) {
-            if (this._type === 'interpolation_box') {
+            if (this._type === 'interpolation_box'
+                || this._type === 'interpolation_points') {
                 counter += window.cvat.player.frames.stop - visibleFrame + 1;
             }
             else {