학교 과제로 헬멧을 감지하는 프로그램을 코딩 중입니다. 파이썬을 통해 프로그램을 실행 시키려고 하는데 작동이 제대로 되지 않고 오류만 뜨네요 해결 방법 좀 알려주세요 ㅜㅜ 이 프로그램은 헬멧을 착용했다면 착용했다는 것을 소리로 알려주고 착용하지 않았다면 착용하지 않았다고 소리로 알려주는 프로그램입니다. 실행 코드는 아래와 같습니다. import cv2 import numpy as np # YOLO 모델 설정 파일 및 가중치 파일 경로 model_cfg = r'C:\Users\USER\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Python 3.12\cfg\yolov4.cfg' model_weights = r'C:\Users\USER\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Python 3.12\weights\yolov4.weights' class_names = r'C:\Users\USER\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Python 3.12\coco.names' # 클래스 이름 로드 with open(class_names, 'r') as f: classes = [line.strip() for line in f.readlines()] # 네트워크 설정 net = cv2.dnn.readNet(model_cfg, model_weights) # 비디오 파일 열기 cap = cv2.VideoCapture('high.mp4') while cap.isOpened(): ret, frame = cap.read() if not ret: break height, width, channels = frame.shape # 네트워크에 이미지를 입력하기 위한 블롭(blob) 생성 blob = cv2.dnn.blobFromImage(frame, 0.00392, (416, 416), (0, 0, 0), True, crop=False) net.setInput(blob) outs = net.forward(net.getUnconnectedOutLayersNames()) # 결과 분석 class_ids = [] confidences = [] boxes = [] for out in outs: for detection in out: scores = detection[5:] class_id = np.argmax(scores) confidence = scores[class_id] if confidence > 0.5: center_x = int(detection[0] * width) center_y = int(detection[1] * height) w = int(detection[2] * width) h = int(detection[3] * height) x = int(center_x - w / 2) y = int(center_y - h / 2) boxes.append([x, y, w, h]) confidences.append(float(confidence)) class_ids.append(class_id) # 노이즈 제거를 위한 Non-Maximum Suppression 적용 indexes = cv2.dnn.NMSBoxes(boxes, confidences, score_threshold=0.5, nms_threshold=0.4) helmet_detected = False for i in indexes.flatten(): x, y, w, h = boxes[i] label = str(classes[class_ids[i]]) if label == 'person': # 'helmet'으로 변경할 수 있음 helmet_detected = True color = (0, 255, 0) # 녹색 cv2.rectangle(frame, (x, y), (x + w, y + h), color, 2) cv2.putText(frame, label, (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 1, color, 2) if helmet_detected: print("헬멧 착용") else: print("헬멧 미착용") cv2.imshow('Frame', frame) if cv2.waitKey(1) & 0xFF == ord('q'): break cap.release() cv2.destroyAllWindows() 이 코딩을 작동 시켰을 때 >>> import cv2 >>> import numpy as np >>> >>> # YOLO 모델 설정 파일 및 가중치 파일 경로 >>> model_cfg = r'C:\Users\USER\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Python 3.12\cfg\yolov4.cfg' >>> model_weights = r'C:\Users\USER\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Python 3.12\weights\yolov4.weights' >>> class_names = r'C:\Users\USER\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Python 3.12\coco.names' >>> >>> # 클래스 이름 로드 >>> with open(class_names, 'r') as f: ... classes = [line.strip() for line in f.readlines()] ... Traceback (most recent call last): File "", line 1, in FileNotFoundError: [Errno 2] No such file or directory: 'C:\\Users\\USER\\AppData\\Roaming\\Microsoft\\Windows\\Start Menu\\Programs\\Python 3.12\\coco.names' >>> # 네트워크 설정 >>> net = cv2.dnn.readNet(model_cfg, model_weights) Traceback (most recent call last): File "", line 1, in cv2.error: OpenCV(4.10.0) D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\darknet\darknet_importer.cpp:210: error: (-212:Parsing error) Failed to open NetParameter file: C:\Users\USER\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Python 3.12\cfg\yolov4.cfg in function 'cv::dnn::dnn4_v20240521::readNetFromDarknet' >>> >>> # 비디오 파일 열기 >>> cap = cv2.VideoCapture('high.mp4') >>> >>> while cap.isOpened(): ... ret, frame = cap.read() ... if not ret: ... break ... >>> height, width, channels = frame.shape File "", line 1 height, width, channels = frame.shape IndentationError: unexpected indent >>> >>> # 네트워크에 이미지를 입력하기 위한 블롭(blob) 생성 >>> blob = cv2.dnn.blobFromImage(frame, 0.00392, (416, 416), (0, 0, 0), True, crop=False) File "", line 1 blob = cv2.dnn.blobFromImage(frame, 0.00392, (416, 416), (0, 0, 0), True, crop=False) IndentationError: unexpected indent >>> net.setInput(blob) File "", line 1 net.setInput(blob) IndentationError: unexpected indent >>> outs = net.forward(net.getUnconnectedOutLayersNames()) File "", line 1 outs = net.forward(net.getUnconnectedOutLayersNames()) IndentationError: unexpected indent >>> >>> # 결과 분석 >>> class_ids = [] File "", line 1 class_ids = [] IndentationError: unexpected indent >>> confidences = [] File "", line 1 confidences = [] IndentationError: unexpected indent >>> boxes = [] File "", line 1 boxes = [] IndentationError: unexpected indent >>> >>> for out in outs: File "", line 1 for out in outs: IndentationError: unexpected indent >>> for detection in out: File "", line 1 for detection in out: IndentationError: unexpected indent >>> scores = detection[5:] File "", line 1 scores = detection[5:] IndentationError: unexpected indent >>> class_id = np.argmax(scores) File "", line 1 class_id = np.argmax(scores) IndentationError: unexpected indent >>> confidence = scores[class_id] File "", line 1 confidence = scores[class_id] IndentationError: unexpected indent >>> if confidence > 0.5: File "", line 1 if confidence > 0.5: IndentationError: unexpected indent >>> center_x = int(detection[0] * width) File "", line 1 center_x = int(detection[0] * width) IndentationError: unexpected indent >>> center_y = int(detection[1] * height) File "", line 1 center_y = int(detection[1] * height) IndentationError: unexpected indent >>> w = int(detection[2] * width) File "", line 1 w = int(detection[2] * width) IndentationError: unexpected indent >>> h = int(detection[3] * height) File "", line 1 h = int(detection[3] * height) IndentationError: unexpected indent >>> x = int(center_x - w / 2) File "", line 1 x = int(center_x - w / 2) IndentationError: unexpected indent >>> y = int(center_y - h / 2) File "", line 1 y = int(center_y - h / 2) IndentationError: unexpected indent >>> >>> boxes.append([x, y, w, h]) File "", line 1 boxes.append([x, y, w, h]) IndentationError: unexpected indent >>> confidences.append(float(confidence)) File "", line 1 confidences.append(float(confidence)) IndentationError: unexpected indent >>> class_ids.append(class_id) File "", line 1 class_ids.append(class_id) IndentationError: unexpected indent >>> >>> # 노이즈 제거를 위한 Non-Maximum Suppression 적용 >>> indexes = cv2.dnn.NMSBoxes(boxes, confidences, score_threshold=0.5, nms_threshold=0.4) File "", line 1 indexes = cv2.dnn.NMSBoxes(boxes, confidences, score_threshold=0.5, nms_threshold=0.4) IndentationError: unexpected indent >>> >>> helmet_detected = False File "", line 1 helmet_detected = False IndentationError: unexpected indent >>> >>> for i in indexes.flatten(): File "", line 1 for i in indexes.flatten(): IndentationError: unexpected indent >>> x, y, w, h = boxes[i] File "", line 1 x, y, w, h = boxes[i] IndentationError: unexpected indent >>> label = str(classes[class_ids[i]]) File "", line 1 label = str(classes[class_ids[i]]) IndentationError: unexpected indent >>> if label == 'person': # 'helmet'으로 변경할 수 있음 File "", line 1 if label == 'person': # 'helmet'으로 변경할 수 있음 IndentationError: unexpected indent >>> helmet_detected = True File "", line 1 helmet_detected = True IndentationError: unexpected indent >>> color = (0, 255, 0) # 녹색 File "", line 1 color = (0, 255, 0) # 녹색 IndentationError: unexpected indent >>> cv2.rectangle(frame, (x, y), (x + w, y + h), color, 2) File "", line 1 cv2.rectangle(frame, (x, y), (x + w, y + h), color, 2) IndentationError: unexpected indent >>> cv2.putText(frame, label, (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 1, color, 2) File "", line 1 cv2.putText(frame, label, (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 1, color, 2) IndentationError: unexpected indent >>> >>> if helmet_detected: File "", line 1 if helmet_detected: IndentationError: unexpected indent >>> print("헬멧 착용") File "", line 1 print("헬멧 착용") IndentationError: unexpected indent >>> else: File "", line 1 else: IndentationError: unexpected indent >>> print("헬멧 미착용") File "", line 1 print("헬멧 미착용") IndentationError: unexpected indent >>> >>> cv2.imshow('Frame', frame) File "", line 1 cv2.imshow('Frame', frame) IndentationError: unexpected indent >>> >>> if cv2.waitKey(1) & 0xFF == ord('q'): File "", line 1 if cv2.waitKey(1) & 0xFF == ord('q'): IndentationError: unexpected indent >>> break File "", line 1 break IndentationError: unexpected indent >>> >>> cap.release() >>> cv2.destroyAllWindows() 이렇게 나옵니다 이런 식으로 작동 오류가 나서 챗gpt를 통해 해결 방법을 물어보니 비디오 파일을 제대로 읽을 수 있는 지 확인을 하라고 import cv2 # 비디오 파일 경로 video_path = r'C:\high\high.mp4' # 분석할 비디오 파일 경로 # 비디오 캡처 설정 cap = cv2.VideoCapture(video_path) # 비디오 파일이 열렸는지 확인 if not cap.isOpened(): print(f"Error: Could not open video file '{video_path}'") exit(1) while cap.isOpened(): ret, frame = cap.read() if not ret: print("Error: Failed to read frame from video file") break # 프레임 표시 cv2.imshow('Video', frame) # 'q' 키를 누르면 종료 if cv2.waitKey(1) & 0xFF == ord('q'): break # 리소스 해제 cap.release() cv2.destroyAllWindows() 이렇게 코딩을 해줘서 실행시키니 파이썬 창이 닫히네요. ㅜㅜ 고수님들 도와주십쇼 ㅜㅜ

코딩테스트 공부 고민....

댓글