-
Notifications
You must be signed in to change notification settings - Fork 54
Description
I have used Deeplabcut GUI to train a model for poses of two rats, which includes three multi-animal-bodyparts and two unique-bodyparts. The model was tested in Deeplab GUI through a video and the result shows that all the 8 points could be dotted.
The main purpose of us is for real-time application, so we find Deeplabcut-Live for next step. We import the model through python code:
deeplabcut.export_model(config_path, shuffle=1, make_tar=False)
Then a new folder named exported-models was created, which included a configuration file pose_cfg.yaml and other model files. Finally, I used deeplabcut-live to process a test video in real time, code:
import cv2
from dlclive import DLCLive, Processor
import time
model_path = r"C:\Users\XHB\Desktop\Multi_Rat-HB-2025-07-31\exported-models\DLC_Multi_Rat_resnet_50_iteration-0_shuffle-1"
dlc_proc = Processor()
dlc_live = DLCLive(model_path, processor=dlc_proc)
cap = cv2.VideoCapture("2rat2.avi")
ret, frame = cap.read()
dlc_live.init_inference(frame)
while True:
if not ret:
break
pose = dlc_live.get_pose(frame)
time_end = time.time()
for x, y, p in pose:
if p > 0.5:
cv2.circle(frame, (int(x), int(y)), 5, (0, 255, 0), -1)
cv2.imshow("DLC Live", frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
The result shows that only 5 points could be dotted, including the 3 multi-animal-bodyparts of the first rat and the 2 unique-bodyparts, while the other 3 multi-animal-bodyparts of the second rat were neglected.
Thus, I wanna ask for help, how could I recognize all the 8 parts of the 2 rats.
configuration file in Deeplabcut GUI:
config.zip
configuration file in Deeplabcut GUI:
pose_cfg.zip

