@@ -307,10 +307,10 @@ def pose_change(self, new_state: Any) -> Any:
307307
308308
309309 group = bp // num_outputs
310- print (f"Group : { group } " )
310+ # print(f"Group : {group}")
311311
312312 other_body_part_indices = [group * num_outputs + i for i in range (num_outputs ) if i != bp % num_outputs ]
313- print (f"Other body part indices: { other_body_part_indices } " )
313+ # print(f"Other body part indices: {other_body_part_indices}")
314314
315315 body_part_is_orig = {bp : False for bp in other_body_part_indices }
316316 old_body_part_data = {} #keep track of these for the undo function
@@ -344,15 +344,15 @@ def pose_change(self, new_state: Any) -> Any:
344344 new_data = SparseTrackingData ()
345345 new_data .pack (* [np .array ([item ]) for item in [y [max_prob_idx ], x [max_prob_idx ], 1 , x_offset [max_prob_idx ], y_offset [max_prob_idx ]]])
346346
347- print (f"Point location: { (x [max_prob_idx ], y [max_prob_idx ])} " )
347+ # print(f"Point location: {(x[max_prob_idx], y[max_prob_idx])}")
348348 for other_bp in other_body_part_indices :
349349 bp_y , bp_x , bp_prob , bp_x_offset , bp_y_offset = copy .deepcopy (frames [frm ][other_bp ].orig_data ).unpack ()
350350 idx = 0
351351 for c_x , c_y in zip (bp_x , bp_y ):
352- print (f"Checking location: { (c_x , c_y )} " )
352+ # print(f"Checking location: {(c_x, c_y)}")
353353 if c_x == x [max_prob_idx ] and c_y == y [max_prob_idx ]:
354- print (f"Found matching location: { (c_x , c_y )} " )
355- print ()
354+ # print(f"Found matching location: {(c_x, c_y)}")
355+ # print()
356356 bp_prob [idx ] = 0
357357
358358 new_bp_data = SparseTrackingData ()
@@ -368,7 +368,7 @@ def pose_change(self, new_state: Any) -> Any:
368368 new_body_part_data [other_bp ] = new_bp_frame
369369
370370 idx += 1
371- print (f"New body part data: { new_body_part_data } " )
371+ # print(f"New body part data: {new_body_part_data}")
372372
373373
374374
@@ -514,7 +514,7 @@ def predict_location(
514514
515515 peak_locs = find_peaks (xs , ys , probs , meta .width )
516516 peak_locs = peak_locs [probs [peak_locs ] >= config .minimum_peak_value ]
517- print (peak_locs )
517+ # print(peak_locs)
518518 if (len (peak_locs ) <= 1 ):
519519 # No peaks, or only one peak, perform basically a no-op, return prior frame state...
520520 x , y , prob = self ._frame_engine .scmap_to_video_coord (
@@ -572,10 +572,10 @@ def pose_change(self, new_state: Any) -> Any:
572572 num_outputs = self ._frame_engine .frame_data .metadata .num_outputs #the number of individuals
573573
574574 group = bp // num_outputs
575- print (f"Group : { group } " )
575+ # print(f"Group : {group}")
576576
577577 other_body_part_indices = [group * num_outputs + i for i in range (num_outputs ) if i != bp % num_outputs ]
578- print (f"Other body part indices: { other_body_part_indices } " )
578+ # print(f"Other body part indices: {other_body_part_indices}")
579579
580580 body_part_is_orig = {bp : False for bp in other_body_part_indices }
581581 old_body_part_data = {} #keep track of these for the undo function
0 commit comments