Skip to content

Commit

Permalink
remove outlier rejection
Browse files Browse the repository at this point in the history
  • Loading branch information
akshay-krishnan committed Aug 11, 2024
1 parent 45f45da commit ad0d2df
Showing 1 changed file with 10 additions and 8 deletions.
18 changes: 10 additions & 8 deletions gtsfm/averaging/translation/averaging_1dsfm.py
Original file line number Diff line number Diff line change
Expand Up @@ -273,7 +273,7 @@ def compute_inliers(
# Compute outlier weights in parallel.
_t2 = timeit.default_timer()
batched_outlier_weights = dask.compute(*batched_outlier_weights)
logger.info("Computed outlier weights using MFAS in %.2f seconds." % (timeit.default_timer() - _t2))
logger.info("Computed outlier weights using MFAS in %.2f seconds.", timeit.default_timer() - _t2)

# Compute average outlier weight.
outlier_weights_sum: DefaultDict[Tuple[int, int], float] = defaultdict(float)
Expand Down Expand Up @@ -529,13 +529,15 @@ def run_translation_averaging(
w_i2Ui1_dict_tracks = {}

inlier_computation_start_time = time.time()
if self._reject_outliers:
w_i2Ui1_dict_inliers, w_i2Ui1_dict_tracks_inliers, inlier_cameras = self.compute_inliers(
w_i2Ui1_dict, w_i2Ui1_dict_tracks
)
else:
w_i2Ui1_dict_inliers = w_i2Ui1_dict
w_i2Ui1_dict_tracks_inliers = w_i2Ui1_dict_tracks
# if self._reject_outliers:
# w_i2Ui1_dict_inliers, w_i2Ui1_dict_tracks_inliers, inlier_cameras = self.compute_inliers(
# w_i2Ui1_dict, w_i2Ui1_dict_tracks
# )
# else:
w_i2Ui1_dict_inliers = w_i2Ui1_dict
w_i2Ui1_dict_tracks_inliers = w_i2Ui1_dict_tracks

logger.info("Running translation averaging on %d unit translations", len(i2Ui1_dict))

inlier_computation_time = time.time() - inlier_computation_start_time

Expand Down

0 comments on commit ad0d2df

Please sign in to comment.