Update localization_log_gamma_model.py

localization_log_gamma_model.py

@@ -55,6 +55,7 @@ def fittingSingle(filepath, args):
     best_env_gamma = None
     best_rsses = None
     for i in range(args.iternum):
+        try:
             result = modelfit_log_gamma(
                 rx_locs,
                 rx_rsses,
@@ -65,6 +66,8 @@ def fittingSingle(filepath, args):
                 monte_carlo_sampling=args.montecarlo,
                 monte_carlo_sampling_rate=args.montecarlo_rate
             )
+        except RuntimeError:
+            continue
         # unpack
         fit_mse, est_tx_loc, est_tx_pwr, est_env_gamma, est_rsses = result
         if fit_mse < min_fit_mse:
@@ -75,6 +78,9 @@ def fittingSingle(filepath, args):
             best_env_gamma = est_env_gamma
             best_rsses = est_rsses
 
+    if best_tx_loc is None:
+        return
+    
     # calculate diff
     data_best_est = convert_vector_to_mat(rx_locs, best_rsses, data.shape)
     data_diff = data - data_best_est