Skip to content

Commit 1900d7a

Browse files
committed
hotfix for dlc issue with pose/peaks
1 parent 00a6eac commit 1900d7a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

DeepLabStream.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -277,15 +277,15 @@ def get_pose_mp(input_q, output_q):
277277
"""
278278

279279
if MODEL_ORIGIN == 'DLC' or MODEL_ORIGIN == 'MADLC':
280+
from utils.poser import find_local_peaks_new
280281
config, sess, inputs, outputs = load_deeplabcut()
281282
while True:
282283
if input_q.full():
283284
index, frame = input_q.get()
284285
if MODEL_ORIGIN == 'DLC':
285286
scmap, locref, pose = get_pose(frame, config, sess, inputs, outputs)
286-
# TODO: Remove alterations to original
287-
#peaks = find_local_peaks_new(scmap, locref, ANIMALS_NUMBER, config)
288-
peaks = pose
287+
peaks = find_local_peaks_new(scmap, locref, ANIMALS_NUMBER, config)
288+
#peaks = pose
289289
if MODEL_ORIGIN == 'MADLC':
290290
peaks = get_ma_pose(frame, config, sess, inputs, outputs)
291291

0 commit comments

Comments
 (0)