added dlc live script

app.py

@@ -27,6 +27,29 @@ def yolo(im, size=640):
     results.render()  # updates results.imgs with boxes and labels
     return Image.fromarray(results.imgs[0])
 
+
+def dlclive_pose(model, crop_np, crop, fname, index,dlc_proc):
+    model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/md_v5a.0.0.pt")
+    
+    dlc_live = DLCLive(model, processor=dlc_proc) 
+    dlc_live.init_inference(crop_np)
+    keypts = dlc_live.get_pose(crop_np) 
+    savetxt(str(fname)+ '_' + str(index) + '.csv' , keypts, delimiter=',')
+    xpose = []
+    ypose = []
+    for key in keypts[:,2]:
+       # if key > 0.05: # which value do we need here?
+            i = np.where(keypts[:,2]==key)
+            xpose.append(keypts[i,0])
+            ypose.append(keypts[i,1])
+    plt.imshow(crop)
+    plt.scatter(xpose[:], ypose[:], 40, color='cyan')
+    plt.savefig(str(fname)+ '_' + str(index) + '.png')
+    plt.show()
+    plt.clf()
+    
+ dlc_proc = Processor()
+    
 #Layouts and descriptions
 title = "MegaDetector and DeepLabcutLive"
 description = "Interact with MegaDetector and DeeplabCutLive for pose estimation"