Skip to content

Commit 98a585a

Browse files
committed
remove debugging outputs
1 parent 0818581 commit 98a585a

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

diplomat/predictors/supervised_fpe/labelers.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)