본문 바로가기

💻 Programming97

[Object Detection] 누구나 쉽게 따라할 수 있는 YOLOv5 모델 학습하기 | 커스텀 데이터 | YOLOv5 예제 코드 안녕하세요! AI & Computer Vision Engineer 뭅즤입니다 👋 본 포스팅은 YOLOv5를 커스텀 데이터셋으로 학습 및 테스트해 보는 예제 코드를 설명하는 글인데요.처음 글을 작성했을 때보다 훨씬 더 쉽게 실습할 수 있도록 수정했으니, 참고 부탁드려요! 🤗📌 Object Detection Object Detection(객체 검출)은 이미지나 동영상 속에 있는 여러 객체의 유형과 위치를 알아내는 기술이에요. 이 기술 덕분에 자율주행 자동차가 주변의 사람이나 차량을 인식하고, 보안 카메라가 의심스러운 물체를 감지하거나, 스마트폰 카메라가 사진 속의 특정 물체에 초점을 맞출 수 있는 거라고 볼 수 있죠! 객체 검출하면 많은 분들이 떠올리는 대표적인 모델이 바로 YOLO 모델일 거예요. .. 2022. 12. 17.

[VS Code] 이미지 관련 Extension 추천 : Image preview, Python Image Preview VS Code엔 정말 편한 extension 플러그인들이 많은데, 그 중에서 이미지 데이터를 자주 다루는 분들께 유용한 플러그인 두 가지를 소개한다. 1. Image preview 이 플러그인을 사용하면 코드를 실행하기 전에 이미지 경로에 마우스를 올리면 이미지 썸네일을 보여준다. 필수적이진 않지만 내가 원하는 이미지 경로를 적었는지 시각적으로 확인할 수 있어서 유용하다. *자세한 사용법은 위 동영상을 참고 2. Python Image Preview 이 플러그인은 1번에 비해 훨씬 더 유용하다. 영상처리 알고리즘이나 딥러닝 모델을 학습하다 보면 이미지가 어떻게 변환되었는지 이미지 데이터가 잘들어갔는지 궁금할 때가 많은데, 그럴 때마다 이미지를 따로 저장해서 열어보기가 굉장히 귀찮다. 하지만 Python .. 2022. 12. 16.

[python] 반복문 효과적으로 디버깅하기: try-except 블록 반복문을 사용하다 보면 예기치 않은 에러가 발생하여 프로그램이 중단되는 경우가 많아요. 이러한 상황에서 Python의 try-except 블록을 활용하면 에러를 효과적으로 처리하고 프로그램의 안정성을 높일 수 있는데요. 이 포스트에서는 try-except 블록을 사용하여 반복문 내부에서 발생하는 에러를 처리하는 방법에 대해 알아볼게요 !왜 try-except 블록을 사용할까요?반복문 내에서 예외가 발생하면 프로그램이 중단되기 때문에, 나머지 반복문을 실행하지 못해요. 예외 처리를 통해 이러한 문제를 방지하고, 예외 상황을 적절히 처리할 수 있습니다. try-except 블록 사용 예제for i in range(10): try: # 에러가 발생할 가능성이 있는 코드 resu.. 2022. 12. 15.

[pytorch] pytorch 모델 로드 중 Missing key(s) in state_dict 에러 pytorch로 딥러닝 모델을 저장하고 불러올 때 Missing key(s) in state_dict 라는 런타임 에러가 발생하는 경우가 종종 있다. 대부분 딥러닝 모델과 불러온 모델 웨이트의 키값이 맞지 않아서 발생하는 오류인데, 모델과 모델 웨이트가 서로 짝이 아닌 경우에 발생하고 간혹 그렇지 않은 경우에도 해당 에러가 발생해서 우리를 괴롭힌다... ㅎ 그 때 아래와 같이 torch.load로 모델의 state dict를 불러오고 디버깅을 해서 state dict의 키값을 확인해보면 정확하게 문제를 파악할 수 있다. model_path = './model.pth' model_state_dict = torch.load(model_path) 위 첫 번째 캡처를 보면 모델 state dict에 모델 웨이트.. 2022. 12. 15.

[openCV] 이미지 위에 선, 사각형, 원 그리기 이미지에서 무언가 검출하고 표시할 때 가장 많이 사용하는 line, rectangle, circle 세 가지 간단 정리. cv2.line(이미지, (시작좌표), (끝 좌표), 컬러, 두께) cv2.rectangle(이미지, (좌상단 좌표), (우하단 좌표), 컬러, 두께) cv2.circle(이미지, (중심 좌표), 반지름, 컬러, 두께) import numpy as np import cv2 black_canvas = np.zeros((500,500,3)) R, G, B = (0,0,255), (0,255,0), (255,0,0) cv2.line(black_canvas, (10,100),(500,300), R, thickness=10) cv2.rectangle(black_canvas, (10, 10), .. 2022. 10. 31.

[python] 주피터 노트북 환경에서 이미지 출력하기 팁 def img_show(title='image', img=None, figsize=(8 ,5)): plt.figure(figsize=figsize) if type(img) == list: if type(title) == list: titles = title else: titles = [] for i in range(len(img)): titles.append(title) for i in range(len(img)): if len(img[i].shape) 2022. 10. 31.

이전 1 ··· 9 10 11 12 13 14 15 ··· 17 다음

티스토리툴바