diff --git a/data/images/pic/test_0.jpg b/data/images/pic/test_0.jpg new file mode 100644 index 0000000..ca82cf1 Binary files /dev/null and b/data/images/pic/test_0.jpg differ diff --git a/data/images/pic/test_10.jpg b/data/images/pic/test_10.jpg new file mode 100644 index 0000000..5585256 Binary files /dev/null and b/data/images/pic/test_10.jpg differ diff --git a/data/images/pic/test_100.jpg b/data/images/pic/test_100.jpg new file mode 100644 index 0000000..f4ec0b6 Binary files /dev/null and b/data/images/pic/test_100.jpg differ diff --git a/data/images/pic/test_110.jpg b/data/images/pic/test_110.jpg new file mode 100644 index 0000000..daf4cc9 Binary files /dev/null and b/data/images/pic/test_110.jpg differ diff --git a/data/images/pic/test_120.jpg b/data/images/pic/test_120.jpg new file mode 100644 index 0000000..8d849ff Binary files /dev/null and b/data/images/pic/test_120.jpg differ diff --git a/data/images/pic/test_130.jpg b/data/images/pic/test_130.jpg new file mode 100644 index 0000000..64aa289 Binary files /dev/null and b/data/images/pic/test_130.jpg differ diff --git a/data/images/pic/test_140.jpg b/data/images/pic/test_140.jpg new file mode 100644 index 0000000..f0ccb3a Binary files /dev/null and b/data/images/pic/test_140.jpg differ diff --git a/data/images/pic/test_150.jpg b/data/images/pic/test_150.jpg new file mode 100644 index 0000000..8e637e4 Binary files /dev/null and b/data/images/pic/test_150.jpg differ diff --git a/data/images/pic/test_160.jpg b/data/images/pic/test_160.jpg new file mode 100644 index 0000000..9c8ea26 Binary files /dev/null and b/data/images/pic/test_160.jpg differ diff --git a/data/images/pic/test_170.jpg b/data/images/pic/test_170.jpg new file mode 100644 index 0000000..4e263b6 Binary files /dev/null and b/data/images/pic/test_170.jpg differ diff --git a/data/images/pic/test_180.jpg b/data/images/pic/test_180.jpg new file mode 100644 index 0000000..27a9ca2 Binary files /dev/null and b/data/images/pic/test_180.jpg differ diff --git a/data/images/pic/test_190.jpg b/data/images/pic/test_190.jpg new file mode 100644 index 0000000..bcdacb0 Binary files /dev/null and b/data/images/pic/test_190.jpg differ diff --git a/data/images/pic/test_20.jpg b/data/images/pic/test_20.jpg new file mode 100644 index 0000000..e8357c2 Binary files /dev/null and b/data/images/pic/test_20.jpg differ diff --git a/data/images/pic/test_200.jpg b/data/images/pic/test_200.jpg new file mode 100644 index 0000000..250646e Binary files /dev/null and b/data/images/pic/test_200.jpg differ diff --git a/data/images/pic/test_210.jpg b/data/images/pic/test_210.jpg new file mode 100644 index 0000000..78ab743 Binary files /dev/null and b/data/images/pic/test_210.jpg differ diff --git a/data/images/pic/test_220.jpg b/data/images/pic/test_220.jpg new file mode 100644 index 0000000..3df2e76 Binary files /dev/null and b/data/images/pic/test_220.jpg differ diff --git a/data/images/pic/test_230.jpg b/data/images/pic/test_230.jpg new file mode 100644 index 0000000..637662d Binary files /dev/null and b/data/images/pic/test_230.jpg differ diff --git a/data/images/pic/test_240.jpg b/data/images/pic/test_240.jpg new file mode 100644 index 0000000..625ab95 Binary files /dev/null and b/data/images/pic/test_240.jpg differ diff --git a/data/images/pic/test_250.jpg b/data/images/pic/test_250.jpg new file mode 100644 index 0000000..37f5118 Binary files /dev/null and b/data/images/pic/test_250.jpg differ diff --git a/data/images/pic/test_260.jpg b/data/images/pic/test_260.jpg new file mode 100644 index 0000000..397c386 Binary files /dev/null and b/data/images/pic/test_260.jpg differ diff --git a/data/images/pic/test_270.jpg b/data/images/pic/test_270.jpg new file mode 100644 index 0000000..adaa490 Binary files /dev/null and b/data/images/pic/test_270.jpg differ diff --git a/data/images/pic/test_280.jpg b/data/images/pic/test_280.jpg new file mode 100644 index 0000000..d4cb5b7 Binary files /dev/null and b/data/images/pic/test_280.jpg differ diff --git a/data/images/pic/test_290.jpg b/data/images/pic/test_290.jpg new file mode 100644 index 0000000..135b621 Binary files /dev/null and b/data/images/pic/test_290.jpg differ diff --git a/data/images/pic/test_30.jpg b/data/images/pic/test_30.jpg new file mode 100644 index 0000000..1307a58 Binary files /dev/null and b/data/images/pic/test_30.jpg differ diff --git a/data/images/pic/test_300.jpg b/data/images/pic/test_300.jpg new file mode 100644 index 0000000..aff1b34 Binary files /dev/null and b/data/images/pic/test_300.jpg differ diff --git a/data/images/pic/test_310.jpg b/data/images/pic/test_310.jpg new file mode 100644 index 0000000..632c54e Binary files /dev/null and b/data/images/pic/test_310.jpg differ diff --git a/data/images/pic/test_320.jpg b/data/images/pic/test_320.jpg new file mode 100644 index 0000000..6ce2cb7 Binary files /dev/null and b/data/images/pic/test_320.jpg differ diff --git a/data/images/pic/test_330.jpg b/data/images/pic/test_330.jpg new file mode 100644 index 0000000..29bca1f Binary files /dev/null and b/data/images/pic/test_330.jpg differ diff --git a/data/images/pic/test_340.jpg b/data/images/pic/test_340.jpg new file mode 100644 index 0000000..d6fa5c2 Binary files /dev/null and b/data/images/pic/test_340.jpg differ diff --git a/data/images/pic/test_40.jpg b/data/images/pic/test_40.jpg new file mode 100644 index 0000000..e74372c Binary files /dev/null and b/data/images/pic/test_40.jpg differ diff --git a/data/images/pic/test_50.jpg b/data/images/pic/test_50.jpg new file mode 100644 index 0000000..e38ea20 Binary files /dev/null and b/data/images/pic/test_50.jpg differ diff --git a/data/images/pic/test_60.jpg b/data/images/pic/test_60.jpg new file mode 100644 index 0000000..fa0e331 Binary files /dev/null and b/data/images/pic/test_60.jpg differ diff --git a/data/images/pic/test_70.jpg b/data/images/pic/test_70.jpg new file mode 100644 index 0000000..6733922 Binary files /dev/null and b/data/images/pic/test_70.jpg differ diff --git a/data/images/pic/test_80.jpg b/data/images/pic/test_80.jpg new file mode 100644 index 0000000..1ce749b Binary files /dev/null and b/data/images/pic/test_80.jpg differ diff --git a/data/images/pic/test_90.jpg b/data/images/pic/test_90.jpg new file mode 100644 index 0000000..e9c57c2 Binary files /dev/null and b/data/images/pic/test_90.jpg differ diff --git a/runs/pose/predict3/test.avi b/runs/pose/predict3/test.avi new file mode 100644 index 0000000..d0cb08d Binary files /dev/null and b/runs/pose/predict3/test.avi differ diff --git a/src/__pycache__/video.cpython-313.pyc b/src/__pycache__/video.cpython-313.pyc index 2113b56..f062c2f 100644 Binary files a/src/__pycache__/video.cpython-313.pyc and b/src/__pycache__/video.cpython-313.pyc differ diff --git a/src/aa.py b/src/label_app.py similarity index 67% rename from src/aa.py rename to src/label_app.py index a7e5f20..2c8f827 100644 --- a/src/aa.py +++ b/src/label_app.py @@ -9,16 +9,18 @@ label = input("请输入标签：") # model = YOLO("yolov8n.pt") model = YOLO(r"C:/workspace/le-yolo/runs/detect/train42/weights/best.pt") -video_path = 'C:/workspace/le-yolo/res/3.mp4' -video_to_pic(video_path) -file_path = 'C:/workspace/le-yolo/data/images/test/' -for filename in os.listdir(file_path): - img = cv2.imread(filename) - results = model(img) - # img_path = '../data/images/train/3fb0f9ac-t2_0.jpg' - # img = cv2.imread(img_path) - # results = model(img) - +video_path = 'C:/workspace/le-yolo/res/2.mp4' +cap = cv2.VideoCapture(video_path) +if not cap.isOpened(): + print("无法打开视频") + exit() +while True: + # 读取一帧 + ret, frame = cap.read() + # 如果读取失败，退出循环 + if not ret: + break + results = model(frame) detections = results[0].boxes boxes = detections.xyxy.cpu().numpy() x1_min = int(np.min(boxes[:, 0])) @@ -26,15 +28,17 @@ for filename in os.listdir(file_path): x2_max = int(np.max(boxes[:, 2])) y2_max = int(np.max(boxes[:, 3])) whole_image_box = np.array([[x1_min, y1_min, x2_max, y2_max]]) - # 可视化 original_boxes = results[0].boxes.xyxy.tolist() new_box = whole_image_box.tolist() combined_boxes = original_boxes + new_box combined_boxes = np.array(combined_boxes) - annotated_image = results[0].plot() cv2.imshow(label, annotated_image) - cv2.waitKey(0) - cv2.destroyAllWindows() + if cv2.waitKey(25) & 0xFF == ord(' '): + break + +cap.release() +cv2.destroyAllWindows() + diff --git a/src/pose.yaml b/src/pose.yaml deleted file mode 100644 index cea9b13..0000000 --- a/src/pose.yaml +++ /dev/null @@ -1,11 +0,0 @@ -path: /home/le/le-yolo/data -train: images/train -val: images/val -test: images/test - -# 关键点坐标 13个关键点，每个关键点有3个坐标[x,y,v] -kpt_shape: [13, 3] - -# 翻转索引 -flip_idx: [5, 4, 3, 2, 1, 0, 8, 9, 6, 7] -names: [ 'person' ] \ No newline at end of file diff --git a/src/pose_show.py b/src/pose_app.py similarity index 96% rename from src/pose_show.py rename to src/pose_app.py index 6d94406..4321e4d 100644 --- a/src/pose_show.py +++ b/src/pose_app.py @@ -10,7 +10,8 @@ cap = cv2.VideoCapture(vide_opath) # cap = cv2.VideoCapture(0) # 打开摄像头 # 连接顺序 （0,1) 0-1 -# 17个关键点 鼻子 眼睛 耳朵 肩膀 手肘 手腕 胯 膝盖 脚腕 +# 17个关键点 +# 鼻子 [眼睛 耳朵 肩膀 手肘 手腕 胯 膝盖 脚腕] connections = [ (3, 1), (1, 0), (0, 2), (2, 4), (1, 2), (4, 6), (3, 5), (5, 6), (5, 7), (7, 9), diff --git a/src/video.py b/src/video.py index 97f2f5c..0e0460e 100644 --- a/src/video.py +++ b/src/video.py @@ -1,5 +1,5 @@ import cv2 -# 视频抽帧 并保存 +# 视频抽帧 并保存 video = cv2.VideoCapture(0) # 摄像头 def video_to_pic(vide_opath): # vide_opath = 'C:/workspace/le-yolo/res/6.mp4' video = cv2.VideoCapture(vide_opath) @@ -9,7 +9,7 @@ def video_to_pic(vide_opath): else: ret = False timeF = 10 # 帧数 - filepath = 'C:/pic/test_' # 保存图片的路径 + filepath = 'C:/workspace/le-yolo/data/images/pic/test_' # 保存图片的路径 while ret: ret, frame = video.read() if num % timeF == 0: