Update eval.py

google-research-datasets · Feb 19, 2021 · aa1604d · aa1604d
1 parent aeee3aa
commit aa1604d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/objectron/dataset/eval.py b/objectron/dataset/eval.py
@@ -38,7 +38,7 @@
 _MAX_PIXEL_ERROR = 20.
 _MAX_AZIMUTH_ERROR = 30.
 _MAX_POLAR_ERROR = 20.
-_MAX_DISTANCE = 0.2   # In meters
+_MAX_DISTANCE = 1.0   # In meters
 _NUM_BINS = 21
 
 def safe_divide(i1, i2):
@@ -172,8 +172,8 @@ def evaluate(self, batch):
           adds = _MAX_DISTANCE
 
         iou_hit_miss.record_hit_miss(iou)
-        add_hit_miss.record_hit_miss(add)
-        adds_hit_miss.record_hit_miss(adds)
+        add_hit_miss.record_hit_miss(add, greater=False)
+        adds_hit_miss.record_hit_miss(adds, greater=False)
         pixel_hit_miss.record_hit_miss(pixel_error, greater=False)
         azimuth_hit_miss.record_hit_miss(azimuth_error, greater=False)
         polar_hit_miss.record_hit_miss(polar_error, greater=False)
@@ -551,4 +551,4 @@ def main(argv):
 if __name__ == '__main__':
   flags.mark_flag_as_required('report_file')
   flags.mark_flag_as_required('eval_data')
-  app.run(main)
+  app.run(main)