xmrt commited on
Commit
11b00b8
·
1 Parent(s): f5eb0df
Files changed (1) hide show
  1. main.py +22 -1
main.py CHANGED
@@ -217,7 +217,28 @@ def run_UI():
217
  webcam_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand position", show_label=True)
218
  webcam_output4 = gr.Video(height=512, label = "Detection and tracking", show_label=True, format="mp4")
219
 
220
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
221
 
222
  # From file
223
  submit_pose_file.click(fn=pose2d,
 
217
  webcam_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand position", show_label=True)
218
  webcam_output4 = gr.Video(height=512, label = "Detection and tracking", show_label=True, format="mp4")
219
 
220
+ with gr.Tab("General information"):
221
+ gr.Markdown("You can load the keypoints in python in the following way: ")
222
+ gr.Code(
223
+ value="""def hello_world():
224
+ return "Hello, world!"
225
+
226
+ print(hello_world())""",
227
+ language="python",
228
+ interactive=True,
229
+ show_label=False,
230
+ )
231
+
232
+ gr.Markdown("""Information about the models
233
+ Pose models: `mmpose` is a library for human pose estimation that provides pre-trained models for 2D and 3D pose estimation.
234
+ The 2D pose model is used for estimating the 2D coordinates of human body joints from an image or a video frame. The model uses a convolutional neural network (CNN) to predict the joint locations and their confidence scores.
235
+ The 2D hand model is a specialized version of the 2D pose model that is designed for hand pose estimation. It uses a similar CNN architecture to the 2D pose model but is trained specifically for detecting the joints in the hand.
236
+ The 3D pose model is used for estimating the 3D coordinates of human body joints from an image or a video frame. The model uses a combination of 2D pose estimation and depth estimation to infer the 3D joint locations.
237
+ All of these models are pre-trained on large datasets and can be fine-tuned on custom datasets for specific applications.
238
+
239
+ Ultralight detection and tracking model: The `track()` method in the Ultralight model is used for object tracking in videos. It takes a video file or a camera stream as input and returns the tracked objects in each frame. The method uses the COCO dataset classes for object detection and tracking. The COCO dataset contains 80 classes of objects such as person, car, bicycle, etc. The `track()` method uses the COCO classes to detect and track the objects in the video frames.
240
+ The tracked objects are represented as bounding boxes with labels indicating the class of the object. The Ultralight model is designed to be fast and efficient, making it suitable for real-time object tracking applications.""")
241
+
242
 
243
  # From file
244
  submit_pose_file.click(fn=pose2d,