BLOG 인공지능
인공지능

Jetson Orin NX에 torch 2.1.0와 torchvision 0.16.1 설치하기

우연한 기회에 Jetson Orin NX를 만져볼 기회가 생겨 간단하게 yolov5를 돌려보고 싶었는데... 그냥 평상시처럼 torch와 torchvision을 설치하고 실행해보니 CUDA가 잡히지 않는 경우, 혹은 아래와 같은 …

Jetson tegra torch torchvision
인공지능

Yolov5 Custom Dataset으로 학습시키고 사용하기

최근에 동아리 프로젝트로 Yolo를 사용할 일이 있었다. yolov5, yolov8은 Colab환경에서 쉽게 학습시킬 수 있는 Pipeline이 제공되고 있어서 쉽게 학습시키고 배포할 수 있었다. 이번에 yolov5를 학습시키고 …

YOLO pytorch
인공지능

Transfer Learning을 이용한 Youtube Thumbnail 이미지 카테고리 분류

이번 포스팅에서는 학교 AI 학회 활동의 일환으로 토이 프로젝트를 진행했던 과정에 대해서 포스팅 해보려고 한다.

.

먼저 주제는 기존 베이스라인 모델을 이용해서 Youtube Thumbnail 이미지의 …

케라스 인공지능 텐서플로 전이학습 이미지분류
인공지능

Attention-based Deep Multiple Instance Learning, 논문 요약과 파이썬 코드

Attention-based Deep Multiple Instance Learning

얼마 전 DACON을 통해 경진대회에 참가하다가 알게 된 논문이다. MIL Problem에 대해서 Pooling을 Attention을 가지고 활용한 알고리즘인데 아주 흥미롭다. 차근차근 …

keras python
인공지능

LightGBM 사용하는 법

Tabular 데이터들을 다룰 때는 보통 딥러닝 보다는 머신러닝 알고리즘을 많이 활용한다. 로지스틱 회귀부터 SVM, 등의 알고리즘이 전통적으로 활용이 많이 되었다. 그러나 요즘에는 부스팅 알고리즘과 앙상블 …

LightGBM 머신러닝 python
Search