File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -154,9 +154,12 @@ def accumulate_value(
154154 average_range : float ,
155155 count : int ,
156156 ):
157- assert count > 0 , f"count must be greater than 0, got { count } "
158- assert np .isfinite (average_epe ), f"average_epe must be finite, got { average_epe } "
159- assert np .isfinite (average_range ), f"average_range must be finite, got { average_range } "
157+ if count == 0 or np .isnan (average_epe ) or np .isnan (average_range ):
158+ print ("Warning in accumulate_value: count is 0 or average_epe/average_range is NaN, skip this entry." )
159+ return
160+ # assert count > 0, f"count must be greater than 0, got {count}"
161+ # assert np.isfinite(average_epe), f"average_epe must be finite, got {average_epe}"
162+ # assert np.isfinite(average_range), f"average_range must be finite, got {average_range}"
160163
161164 class_idx = self .class_names .index (class_name )
162165 range_bucket_idx = self .range_buckets .index (range_bucket )
You can’t perform that action at this time.
0 commit comments