본문 바로가기 메뉴 바로가기

지속 가능한 꾸준함

프로필사진
  • 글쓰기
  • 관리
  • 태그
  • 방명록
  • RSS

지속 가능한 꾸준함

검색하기 폼
  • 분류 전체보기 (704)
    • Math (19)
      • Linear Algebra (18)
      • Probability (0)
    • 언어의 마술사 (2)
    • Deep Learning (143)
      • PyTorch (34)
    • Problem Solving (39)
      • 백준 온라인 저지 (33)
      • 프로그래머스 (1)
      • Softeer (2)
    • IDE (5)
    • Python (54)
    • Git (12)
    • 인생 (39)
    • 영어 (53)
    • Docker (11)
    • CMake (8)
    • Contribution 일지 (39)
    • C++ (22)
    • CUDA (5)
    • 잡담 (9)
    • QT (3)
    • 기타 (189)
    • 맛집 (0)
    • Github Actions (2)
    • 가슴이 웅장해지는 인터뷰 (1)
    • 노래 (0)
    • Computer Vision (4)
  • 방명록

C++ Deploy (1)
[PyTorch & LibTorch] OpenCV Image to Normalized Tensor C++ 코드

1. 이미지를 Load 한다. 2. 이미지를 Resize 한다. 3. (모델이 요구하는 인풋 텐서의 컬러 포맷이 RGB인 경우) 이미지의 컬러 포맷을 RGB로 변환한다. *OpenCV는 기본적으로 BGR 포맷으로 이미지를 처리한다. 4. 이미지를 텐서(torch::Tensor 형)로 변환한다. 5. 모델이 요구하는 인풋 텐서의 Shape에 맞도록 텐서를 Reshape한다. (본 예제에서는 NCHW 포맷을 따름) 6. 텐서를 정규화한다. 7. (GPU를 사용하는 경우) 텐서를 GPU 메모리로 업로드한다. cv::Mat img_bgr_u8 = cv::imread("your_image.jpg",cv::IMREAD_COLOR); cv::Mat img_rgb_u8; //resize cv::resize(img_bg..

Deep Learning/PyTorch 2021. 2. 18. 15:31
이전 1 다음
이전 다음
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
TAG
  • PyCharm
  • cosine
  • 백준
  • 백준 11437
  • 순열
  • C++ Deploy
  • 백트래킹
  • 자료구조
  • 인공지능을 위한 선형대수
  • 문제집
  • 파이참
  • LCA
  • 가장 긴 증가하는 부분 수열
  • 위상 정렬 알고리즘
  • FairMOT
  • 단축키
  • ㅂ
  • MOT
  • 조합
  • 이분탐색
  • 백준 1766
  • 백준 11053
  • Lowest Common Ancestor
more
«   2025/12   »
일 월 화 수 목 금 토
1 2 3 4 5 6
7 8 9 10 11 12 13
14 15 16 17 18 19 20
21 22 23 24 25 26 27
28 29 30 31
글 보관함

Blog is powered by Tistory / Designed by Tistory

티스토리툴바