yolov5でのUSBカメラの使用方法について

Question

前提・実現したいこと

USB接続のカメラを用いて、yolov5の動体検知がしたいです。

発生している問題・エラーメッセージ

Namespace(agnostic_nms=False, augment=False, classes=None, conf_thres=0.25, device='', exist_ok=False, img_size=640, iou_thres=0.45, name='exp', project='runs/detect', save_conf=False, save_txt=False, source='0', update=False, view_img=False, weights='yolov5s.pt')
Using torch 1.7.1 CPU

Fusing layers... 
Model Summary: 232 layers, 7459581 parameters, 0 gradients, 17.5 GFLOPS
1/1: 0...  success (1280x960 at 25.00 FPS).

Traceback (most recent call last):
  File "detect.py", line 174, in <module>
    detect()
  File "detect.py", line 49, in detect
    dataset = LoadStreams(source, img_size=imgsz)
  File "/Users/slax22/yolov5/utils/datasets.py", line 286, in __init__
    s = np.stack([letterbox(x, new_shape=self.img_size)[0].shape for x in self.imgs], 0)  # inference shapes
  File "/Users/slax22/yolov5/utils/datasets.py", line 286, in <listcomp>
    s = np.stack([letterbox(x, new_shape=self.img_size)[0].shape for x in self.imgs], 0)  # inference shapes
  File "/Users/slax22/yolov5/utils/datasets.py", line 696, in letterbox
    shape = img.shape[:2]  # current shape [height, width]
AttributeError: 'NoneType' object has no attribute 'shape'
### 該当のソースコード
```python
ソースコード
python detect.py --source 0
### 試したこと

--source 0 の0の部分を1や2に変えましたが変化はありませんでした。
またmacを使っているのですが、システムレポートを確認したところUSBカメラはカメラ0として認識されていることがわかっています。


### 補足情報（FW/ツールのバージョンなど）
https://qiita.com/SatoshiGachiFujimoto/items/cb586b255eb220fdd8aa
上記URLに書いてあったことをそのまま実行しているのですが、

SnapCameraなどにウェブカメラを占有される場合は、utils/datasets.pyのLoadStreamsクラス237行目の
cap = cv2.VideoCapture(0 if s == '0' else s)
の0のところを1などに変えてみてください。

この文章がいまいち理解できません。具体的にどのプログラムにどのようなことをすればいいのでしょうか。

Answer 1

yolov5のフォルダーの中の、utilsフォルダーにあるdatasets.pyを開いて、
datasets.pyの275行目に書いてある行のコードを消して

cap = cv2.VideoCapture(1 if s == '0' else s)

に書き換えろってことだと思います。（インデントも忘れないでください。）

Comments

[slax22]

回答誠にありがとうございます。早速試したところ、うまくいきました！ありがとうございました！

[children]

返信遅くなりすみません。初めてベストアンサーをいただいたのでとても嬉しいです。こちらこそ、ありがとうございました！