-
질문 & 답변
카테고리
-
세부 분야
컴퓨터 비전
-
해결 여부
미해결
video inference 오류가 생겨 질문 드립니다
21.11.22 13:12 작성 조회수 211
0
안녕하세요, Ultralytics Yolo 실습 - 02 강의를 따라 인크레더블 데이터 셋 대신 제가 제작한 데이터로 학습을 수행해 video inference 진행 중 오류가 발생해 질문드립니다.
video inference와 image inference 수행 중 다음과 같은 오류가 발생했습니다.
video inference 코드:
# video 파일 inference
cd /content/yolov3;python detect.py --source /content/simult/simul_1.mp4 \
--weights /mydrive/ultra_workdir/simult/weights/best.pt --conf 0.3 \
--project=/content/data/output --name=run_video --exist-ok --line-thickness 2
실행 중 오류 코드:
detect: weights=['/mydrive/ultra_workdir/simult/weights/best.pt'], source=/content/simult/simul_1.mp4, imgsz=[640, 640], conf_thres=0.3, iou_thres=0.45, max_det=1000, device=, view_img=False, save_txt=False, save_conf=False, save_crop=False, nosave=False, classes=None, agnostic_nms=False, augment=False, visualize=False, update=False, project=/content/data/output, name=run_video, exist_ok=True, line_thickness=2, hide_labels=False, hide_conf=False, half=False, dnn=False YOLOv3 🚀 v9.6.0-3-gb870de5 torch 1.10.0+cu111 CUDA:0 (Tesla P100-PCIE-16GB, 16281MiB) Fusing layers... Model Summary: 261 layers, 61497430 parameters, 0 gradients, 154.7 GFLOPs Traceback (most recent call last): File "detect.py", line 244, in <module> main(opt) File "detect.py", line 239, in main run(**vars(opt)) File "/usr/local/lib/python3.7/dist-packages/torch/autograd/grad_mode.py", line 28, in decorate_context return func(*args, **kwargs) File "detect.py", line 149, in run s += f"{n} {names[int(c)]}{'s' * (n > 1)}, " # add to string KeyError: 0
image inference 코드:
# image 파일 inference
!cd /content/yolov3;python detect.py --source /content/simult/images/train/13.png \
--weights /mydrive/ultra_workdir/simult/weights/best.pt --conf 0.3 \
--project=/content/data/output --name=run_image --exist-ok --line-thickness 2
실행 중 오류 코드:
detect: weights=['/mydrive/ultra_workdir/simult/weights/best.pt'], source=/content/simult/images/train/13.png, imgsz=[640, 640], conf_thres=0.3, iou_thres=0.45, max_det=1000, device=, view_img=False, save_txt=False, save_conf=False, save_crop=False, nosave=False, classes=None, agnostic_nms=False, augment=False, visualize=False, update=False, project=/content/data/output, name=run_image, exist_ok=True, line_thickness=2, hide_labels=False, hide_conf=False, half=False, dnn=False YOLOv3 🚀 v9.6.0-3-gb870de5 torch 1.10.0+cu111 CUDA:0 (Tesla P100-PCIE-16GB, 16281MiB) Fusing layers... Model Summary: 261 layers, 61497430 parameters, 0 gradients, 154.7 GFLOPs Traceback (most recent call last): File "detect.py", line 244, in <module> main(opt) File "detect.py", line 239, in main run(**vars(opt)) File "/usr/local/lib/python3.7/dist-packages/torch/autograd/grad_mode.py", line 28, in decorate_context return func(*args, **kwargs) File "detect.py", line 149, in run s += f"{n} {names[int(c)]}{'s' * (n > 1)}, " # add to string KeyError: 0
이러한 오류를 어떻게 해결해야 하는지 알고 싶습니다.
감사합니다.
답변을 작성해보세요.
0
권 철민
지식공유자2021.11.22
안녕하십니까,
저도 에러메시지만 봐서는 정확히 문제가 뭔지 잘 모르겠습니다.
혹시 detect.py를 수정하신게 있나요?
그리고 오류 메시지를 한줄로 하지 마시고, 출력되는 형태대로 올려 주시겠습니까, 메시지를 보기가 어렵군요.
답변 1