Custom

AI

SegFormer custom data로 학습시키기!

segformer의 offical github를 가져와서 custom data로 학습시키는 것을 해보겠다 0. 우선 기본적인 세팅은 끝마쳤다고 생각하겠다 (apt update, upgrade, conda 또는 docker를 이용해 환경 설정 등) https://daeun-computer-uneasy.tistory.com/126 이 링크 참조하시면 좋을 것 같습니다. mmcv 같은 경우, cuda 및 torch 버전에 엄청난 영향을 받아서,, 꼬일 대로 꼬이면 답도 없습니다 ㅜㅜ [PyTorch] mmcv 설치하기 / cuda 버전에 맞게 mmcv downgrade하기 / mmcv._ext error 해결 오늘은 잊을만하면 나오는 mmcv 설치 방법에 대해 총정리하는 포스팅을 해보려고 한다. 사실 항상 설..

파이썬

[Pytorch] custom dataset의 mean, std 구하기

import numpy as np import matplotlib.pyplot as plt from pycocotools.coco import COCO import torch from torchvision import datasets from tqdm.notebook import tqdm def image_stats(data): total_images = len(data) total_pixels = 0 pixel_sum = np.zeros(3) pixel_squared_sum = np.zeros(3) for i in range(total_images): # 이미지 데이터를 numpy 객체로 변환하고 정규화(0~1) img, _ = data[i] img_np = np.asarray(img) / 255.0 ..

AI

Yolo v3 custom data 하나부터 열까지(Windows)

yolo 개발환경 ubuntu 18.0.4 Lts CPU Xserver ​ 0. OPENCV 설치 // 기존 opencv 삭제 sudo apt-get remove libopencv* sudo apt-get autoremove sudo find /usr/local/ -name "*opencv*" -exec rm {} \; // 아무것도 안뜨면 정상 // 업그레이드 sudo apt-get update sudo apt-get upgrade // 개발자 도구 및 각종 라이브러리 설치 sudo apt-get install build-essential cmake unzip pkg-config //for compiling sudo apt-get install libjpeg-dev libpng-dev libtiff-d..

beomseok99
'Custom' 태그의 글 목록