Added basic stats

Author: jsbroks

backend/webserver/api/datasets.py

@@ -134,6 +134,39 @@ def get(self, dataset_id):
         return query_util.fix_ids(users)
 
 
+@api.route('/<int:dataset_id>/stats')
+class DatasetStats(Resource):
+
+    @login_required
+    def get(self, dataset_id):
+        """ All users in the dataset """
+        args = dataset_generate.parse_args()
+
+        dataset = current_user.datasets.filter(id=dataset_id, deleted=False).first()
+        if dataset is None:
+            return {"message": "Invalid dataset id"}, 400
+
+        images = ImageModel.objects(dataset_id=dataset.id, deleted=False)
+        annotations = AnnotationModel.objects(dataset_id=dataset_id, deleted=False)
+        stats = {
+            'total': {
+                'Users': dataset.get_users().count(),
+                'Images': images.count(),
+                'Annotated Images': images.filter(annotated=True).count(),
+                'Annotations': annotations.count(),
+                'Categories': len(dataset.categories)
+            },
+            'average': {
+                'Image Width (px)': images.average('width'),
+                'Image Height (px)': images.average('height'),
+                'Annotation Area (px)': annotations.average('area'),
+                'Time (ms) per Image': images.average('milliseconds') or 0,
+                'Time (ms) per Annotation': annotations.average('milliseconds') or 0
+            }
+        }
+        return stats
+
+
 @api.route('/<int:dataset_id>')
 class DatasetId(Resource):
     @login_required

client/src/models/datasets.js

@@ -51,5 +51,8 @@ export default {
   },
   getUsers(id) {
     return axios.get(`${baseURL}/${id}/users`);
+  },
+  getStats(id) {
+    return axios.get(`${baseURL}/${id}/stats`);
   }
 };

client/src/views/Dataset.vue

@@ -64,12 +64,12 @@
         <div class="container" v-show="tab == 'exports'">Empty</div>
         <div class="container" v-show="tab == 'members'">
 
-          <div class="my-3 p-3 bg-white rounded shadow-sm">
+          <div class="card my-3 p-3 shadow-sm mr-2">
             <h6 class="border-bottom border-gray pb-2"><b>Invite Members</b></h6>
 
           </div>
 
-          <div class="my-3 p-3 bg-white rounded shadow-sm">
+          <div class="card my-3 p-3 shadow-sm mr-2">
             <h6 class="border-bottom border-gray pb-2"><b>Exisiting Memebers</b></h6>
 
             <div class="media text-muted pt-3" v-for="user in users">
@@ -87,7 +87,34 @@
           </div>
 
         </div>
-        <div class="container" v-show="tab == 'statistics'">Stats</div>
+        <div class="container" v-show="tab == 'statistics'">
+          <div v-if="stats == null">
+            Crunching numbers...
+          </div>
+
+          <div v-else>
+            <div class="row">
+              
+              <div v-if="stats.total" class="card my-3 p-3 shadow-sm col-3 mr-2">
+                <h6 class="border-bottom border-gray pb-2"><b>Total</b></h6>
+                <div class="row" v-for="stat in Object.keys(stats.total)">
+                  <strong class="col-8">{{stat}}:</strong>
+                  <span class="col-4">{{stats.total[stat]}}</span>
+                </div>
+              </div>
+
+              <div v-if="stats.average" class="card my-3 p-3 shadow-sm col-4 mr-2">
+                <h6 class="border-bottom border-gray pb-2"><b>Average</b></h6>
+                <div class="row" v-for="stat in Object.keys(stats.average)">
+                  <strong class="col-8">{{stat}}:</strong>
+                  <span class="col-4">{{stats.average[stat].toFixed(2)}}</span>
+                </div>
+              </div>
+
+            </div>
+            
+          </div>
+        </div>
         <div class="container" v-show="tab == 'settings'">settings</div>
 
       </div>
@@ -369,7 +396,8 @@ export default {
       panel: {
         showAnnotated: true,
         showNotAnnotated: true
-      }
+      },
+      stats: null
     };
   },
   methods: {
@@ -419,6 +447,12 @@ export default {
           this.users = response.data
         });
     },
+    getStats() {
+      Dataset.getStats(this.dataset.id)
+        .then(response => {
+          this.stats = response.data;
+        });
+    },
     createScanTask() {
       if (this.scan.id != null) {
         this.$router.push({ path: "/tasks", query: { id: this.scan.id } });
@@ -549,6 +583,7 @@ export default {
     tab(tab) {
       localStorage.setItem('dataset/tab', tab);
       if (tab == 'members') this.getUsers();
+      if (tab == 'statistics') this.getStats();
     },
     queryAnnotated() {
       this.updatePage();
@@ -671,10 +706,4 @@ export default {
   padding: 0 5px 2px 5px;
   overflow: auto;
 }
-
-.card-header {
-  padding: 5px 10px;
-}
-
-
 </style>