File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -147,9 +147,12 @@ def accumulate_value(
147147 average_range : float ,
148148 count : int ,
149149 ):
150- assert count > 0 , f"count must be greater than 0, got { count } "
151- assert np .isfinite (average_epe ), f"average_epe must be finite, got { average_epe } "
152- assert np .isfinite (average_range ), f"average_range must be finite, got { average_range } "
150+ if count == 0 or np .isnan (average_epe ) or np .isnan (average_range ):
151+ print ("Warning in accumulate_value: count is 0 or average_epe/average_range is NaN, skip this entry." )
152+ return
153+ # assert count > 0, f"count must be greater than 0, got {count}"
154+ # assert np.isfinite(average_epe), f"average_epe must be finite, got {average_epe}"
155+ # assert np.isfinite(average_range), f"average_range must be finite, got {average_range}"
153156
154157 class_idx = self .class_names .index (class_name )
155158 range_bucket_idx = self .range_buckets .index (range_bucket )
You can’t perform that action at this time.
0 commit comments