Skip to content

Commit aa1604d

Browse files
authored
Update eval.py
1 parent aeee3aa commit aa1604d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

objectron/dataset/eval.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
_MAX_PIXEL_ERROR = 20.
3939
_MAX_AZIMUTH_ERROR = 30.
4040
_MAX_POLAR_ERROR = 20.
41-
_MAX_DISTANCE = 0.2 # In meters
41+
_MAX_DISTANCE = 1.0 # In meters
4242
_NUM_BINS = 21
4343

4444
def safe_divide(i1, i2):
@@ -172,8 +172,8 @@ def evaluate(self, batch):
172172
adds = _MAX_DISTANCE
173173

174174
iou_hit_miss.record_hit_miss(iou)
175-
add_hit_miss.record_hit_miss(add)
176-
adds_hit_miss.record_hit_miss(adds)
175+
add_hit_miss.record_hit_miss(add, greater=False)
176+
adds_hit_miss.record_hit_miss(adds, greater=False)
177177
pixel_hit_miss.record_hit_miss(pixel_error, greater=False)
178178
azimuth_hit_miss.record_hit_miss(azimuth_error, greater=False)
179179
polar_hit_miss.record_hit_miss(polar_error, greater=False)
@@ -551,4 +551,4 @@ def main(argv):
551551
if __name__ == '__main__':
552552
flags.mark_flag_as_required('report_file')
553553
flags.mark_flag_as_required('eval_data')
554-
app.run(main)
554+
app.run(main)

0 commit comments

Comments
 (0)