.py 형식 파일을 불러와서 실행하는데 요류가 계속 뜹니다.

Question

(base) D:\2022\11\MS	est>age.pyTraceback (most recent call last):File &#34;D:\2022\11\MS	est\age.py&#34;, line 17, in <module>from models.experimental import attempt_loadModuleNotFoundError: No module named 'models'이렇게 오류가 뜹니다.age.py 파일은import osos.chdir('D:/2022/11/MS/test/yolov5')import torchimport cv2import numpy as npfrom models.experimental import attempt_loadfrom utils.general import non_max_suppressiondevice &#61; torch.device(&#34;cuda&#34; if torch.cuda.is_available() else&#34;cpu&#34;)class OBJ_DETECTION(): def __init__(self, model_path, classes): self.classes &#61; classes self.yolo_model &#61; attempt_load(weights&#61;model_path, device&#61;device) self.input_width &#61; 320 def detect(self, main_img): height, width &#61; main_img.shape[:2] new_height &#61; int((((self.input_width / width) height) // 32) 32) img &#61; cv2.resize(main_img, (self.input_width, new_height)) img &#61; cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img &#61; np.moveaxis(img, -1, 0) img &#61; torch.from_numpy(img).to(device) img &#61; img.float() / 255.0 # 0 - 255 to 0.0 - 1.0 if img.ndimension() &#61;&#61; 3: img &#61; img.unsqueeze(0) pred &#61; self.yolo_model(img, augment&#61;False)[0] pred &#61; non_max_suppression(pred, conf_thres&#61;0.25, iou_thres&#61;0.45, classes&#61;None) items &#61; [] if pred[0] is not None and len(pred): for p in pred[0]: score &#61; np.round(p[4].cpu().detach().numpy(), 2) label &#61; self.classes[int(p[5])] xmin &#61; int(p[0] * main_img.shape[1] / self.input_width) ymin &#61; int(p[1] * main_img.shape[0] / new_height) xmax &#61; int(p[2] * main_img.shape[1] / self.input_width) ymax &#61; int(p[3] * main_img.shape[0] / new_height) item &#61; {'label': label, 'bbox': [(xmin, ymin), (xmax, ymax)], 'score': score } items.append(item) return items입니다.해결방법을 알려주시면 좋을꺼 같습니다!

HJOW · Accepted Answer

YOLO 세팅 하셨나요?https://github.com/ultralytics/yolov5그리고 이 질문글도 있는데 답변대로도 해보시구요https://stackoverflow.com/questions/73173192/modulenotfounderror-no-module-named-models

.py 형식 파일을 불러와서 실행하는데 요류가 계속 뜹니다.

1개의 답변

#yolov5 목록