我得到一个错误,而运行此代码片段,我看代码好像无法创建numpy数组,我想它正在运行一个无限循环

wwwo4jvm  于 2023-06-23  发布在  其他
关注(0)|答案(1)|浏览(121)
  1. cap = cv2.VideoCapture(0)
  2. # Set mediapipe model
  3. with mp_holistic.Holistic(min_detection_confidence=0.5, min_tracking_confidence=0.5) as holistic:
  4. # NEW LOOP
  5. # Loop through actions
  6. for action in actions:
  7. # Loop through sequences aka videos
  8. for sequence in range(start_folder, start_folder+no_sequences):
  9. # Loop through video length aka sequence length
  10. for frame_num in range(sequence_length):
  11. # Read feed
  12. ret, frame = cap.read()
  13. cv2.startWindowThread()
  14. # Make detections
  15. image, results = mediapipe_detection(frame, holistic)
  16. # Draw landmarks
  17. draw_styled_landmarks(image, results)
  18. # NEW Apply wait logic
  19. if frame_num == 0:
  20. cv2.putText(image, 'STARTING COLLECTION', (120,200),
  21. cv2.FONT_HERSHEY_SIMPLEX, 1, (0,255, 0), 4, cv2.LINE_AA)
  22. cv2.putText(image, 'Collecting frames for {} Video Number {}'.format(action, sequence), (15,12),
  23. cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 1, cv2.LINE_AA)
  24. # Show to screen
  25. cv2.imshow('OpenCV data collection Feed', image)
  26. cv2.waitKey(1)
  27. else:
  28. cv2.putText(image, 'Collecting frames for {} Video Number {}'.format(action, sequence), (15,12),
  29. cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 1, cv2.LINE_AA)
  30. # Show to screen
  31. cv2.imshow('OpenCV data collection Feed', image)
  32. # NEW Export keypoints
  33. keypoints = extract_keypoints(results)
  34. npy_path = os.path.join(DATA_PATH, action, str(sequence), str(frame_num))
  35. np.save(npy_path, keypoints)
  36. # break
  37. if cv2.waitKey(10) & 0xFF == ord('q'):
  38. break
  39. cap.release()
  40. cv2.destroyAllWindows()
  41. cv2.waitKey(1)

使用OpenCV和MediaPipe库进行数据收集的代码单元。它使用cv2.VideoCapture(0)从默认摄像机(索引0)设置视频捕获,并初始化MediaPipe整体模型。
然后,代码进入一个循环,循环遍历一系列操作。在每个动作中,它在一系列序列上循环,在每个序列中,它在帧上循环。对于每一帧,它使用www.example.com()读取视频源cap.read,并使用mediapipe_detection()函数执行检测。然后使用draw_styled_landmarks()在帧上绘制地标。
我得到的错误是

  1. FileNotFoundError Traceback (most recent call last)
  2. Cell In[58], line 43
  3. 41 keypoints = extract_keypoints(results)
  4. 42 npy_path = os.path.join(DATA_PATH, action, str(sequence), str(frame_num))
  5. ---> 43 np.save(npy_path, keypoints)
  6. 45 # break
  7. 46 if cv2.waitKey(10) & 0xFF == ord('q'):
  8. File <__array_function__ internals>:180, in save(*args, **kwargs)
  9. File ~/miniconda3/envs/tensorflow/lib/python3.10/site-packages/numpy/lib/npyio.py:518, in save(file, arr, allow_pickle, fix_imports)
  10. 516 if not file.endswith('.npy'):
  11. 517 file = file + '.npy'
  12. --> 518 file_ctx = open(file, "wb")
  13. 520 with file_ctx as fid:
  14. 521 arr = np.asanyarray(arr)
  15. FileNotFoundError: [Errno 2] No such file or directory: '/Users/utkx2/Desktop/python/MLDL/Projects/Human-Action-Recognition/MP_Data/hello/31/0.npy'

我的文件夹用于存储数据已经创建,但我不知道为什么它无法创建numpy文件用于存储numpy数组值。亲切的帮助。我使用的是M1 Mac(仅供参考,以防出现操作系统问题或其他问题)

xlpyo6sf

xlpyo6sf1#

我认为你需要创建保存numpy数组的完整路径,更具体地说,你需要在定义的路径中创建所有子文件夹:

  1. npy_path = os.path.join(DATA_PATH, action, str(sequence), str(frame_num))
  2. np.save(npy_path, keypoints)

因此,您可以在调用www.example.com之前调用os.makedirs(npy_path,exit_ok=True)np.save,如果文件夹已经创建,则参数exit_ok=true将不会创建它们。

  1. npy_path = os.path.join(DATA_PATH, action, str(sequence), str(frame_num))
  2. os.makedirs(npy_path, exit_ok=True)
  3. np.save(npy_path, keypoints)

相关问题