250x250
꾸준희
Enough is not enough
꾸준희
전체 방문자
2,319,223
오늘
157
어제
774

공지사항

  • 꾸준희 블로그
  • 분류 전체보기 (580)
    • 생각 (14)
    • Book Review (38)
    • Paper Review (18)
    • Research Topic (126)
      • Deep Learning (24)
      • Pose Estimation (29)
      • Object Detection (22)
      • Object Segmentation (3)
      • Object Tracking (11)
      • Video Surveillance (4)
      • Action Recognition (6)
      • Stereo Vision (6)
      • 3D Reconstruction (4)
      • Machine Vision (2)
      • Image Processing (11)
      • Dataset (4)
    • Development (76)
      • NVIDIA DeepStream (3)
      • NVIDIA TensorRT (30)
      • NVIDIA TAO Toolkit (2)
      • ONNX (8)
      • PyTorch (5)
      • TensorFlow (15)
      • TensorFlow Lite (1)
      • GPU | CUDA | PyCUDA (12)
    • Programming (147)
      • GStreamer | FFmpeg (6)
      • Python (27)
      • C | C++ (15)
      • OpenCV (34)
      • Linux (36)
      • Embedded linux (7)
      • Etc. (22)
    • Computer Science (62)
      • 학부 및 대학원 과목 (22)
      • 선형대수학 및 기타 수학 (7)
      • SQL-D (33)
    • 삽질 기록 (49)
    • 기타 (50)
      • 참고자료 (30)
      • 좋은 글 (5)
      • 티스토리 (2)
      • 논문 작성 관련 참고 (10)
      • 메모 (0)

블로그 메뉴

  • 👀 CV
  • 🌸 GitHub
  • 💌 LinkedIn
  • 📚 방명록

최근 댓글

  • python으로 만든 코드를 이제 a⋯
    android 초보
  • cannot resolve symbol 'type'⋯
    android 초보
  • 정확히 뭐라고 뜨나요?
    꾸준희
  • 앗 ㅠ 제가 오랜만에 스킨을 바⋯
    꾸준희
  • 선생님 ㅜㅠ 각도 계산하는 부⋯
    android 초보
07-04 07:57

티스토리

hELLO · Designed By 정상우.
꾸준희

Enough is not enough

삽질 기록

Image Classification 시 분류 결과 한쪽에 치우쳐져서 나오는 현상

2021. 7. 1. 10:29
반응형

 

weight 파일을 load 하지 않고 테스트 시

model 구조만 불러와서 inferece 하기 때문에

분류 결과는 class A : 0.54, class B : 0.46 이런식의 값이 산출되게 된다.

 

반드시 weight 파일을 load 할 때 path 를 잘 지정해주어야 하며, (예외 처리도 반드시 해야함)

load_state_dict(checkpoint['state_dict'] 을 꼭 해주어야한다. 

반응형
저작자표시비영리
  • 카카오스토리
  • 트위터
  • 페이스북

'삽질 기록' 카테고리의 다른 글

mmdetection 으로부터 학습한 모델을 ONNX 및 TensorRT 로 변환 시 나타나는 문제  (0) 2021.07.26
fatal error: libpq-fe.h: No such file or directory  (0) 2021.07.15
Image Classification 시 분류 결과 한쪽에 치우쳐져서 나오는 현상  (0) 2021.07.01
Linux 환경에서 OpenCV의 VideoCapture 사용 시 Webcam이 죽어도 읽히지 않을 때  (0) 2021.06.25
Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same  (0) 2021.06.24
Ubuntu 18.04 환경에서 VNC Server 설정했을 때 회색 화면 나타나는 문제  (0) 2021.06.11
    '삽질 기록' 카테고리의 다른 글
    • mmdetection 으로부터 학습한 모델을 ONNX 및 TensorRT 로 변환 시 나타나는 문제
    • fatal error: libpq-fe.h: No such file or directory
    • Linux 환경에서 OpenCV의 VideoCapture 사용 시 Webcam이 죽어도 읽히지 않을 때
    • Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same
    꾸준희
    꾸준희
    생각과 기록 그리고 발전
    댓글쓰기
    다음 글
    fatal error: libpq-fe.h: No such file or directory
    이전 글
    [Linux] Code Blocks 설치하기
    • 이전
    • 1
    • ···
    • 64
    • 65
    • 66
    • 67
    • 68
    • 69
    • 70
    • 71
    • 72
    • ···
    • 580
    • 다음

    티스토리툴바