본문 바로가기

pytorch13

[pytorch, 이미지분석] CustomDataset 제작시 주의해야할 점. CustomDataset을 제작하는데 있어서 익숙한 포멧은 다음과 같습니다. import os import pandas as pd from torchvision.io import read_image class CustomImageDataset(Dataset): def __init__(self, annotations_file, img_dir, transform=None, target_transform=None): self.img_labels = pd.read_csv(annotations_file) self.img_dir = img_dir self.transform = transform self.target_transform = target_transform def __len__(self): return l.. 2022. 8. 5.
[pytorch, 딥러닝] 모델 저장하는 방법 최근 논문을 보면서 github에 공개된 코드를 해석하는 중에 공개된 모델을 사용해보고자 하였습니다. 처음에는 제가 배운 모델 로드 방법이 먹히질 않아서(kaggle에서 얻게된 모델 로드방법...) 정말 오랫동안 알아 보는 도중에 알아내게 되었습니다. 그걸 포스팅 해보고자 합니다. 머리카락 한움쿰정도 뽑아진것 같네요. 제가 kaggle에서 배운 모델 저장 방법은 checkpoint를 저장하는 방법이고 모델을 제작할때 사용된 정보들을 이용하여 dictionary를 제작 후에 저장하는 방법입니다. checkpoint = { 'epoch': epoch + 1, 'loss': epoch_val_loss, 'state_dict': model.state_dict(), 'optimizer': optimizer.sta.. 2022. 7. 28.
[pytorch, 딥러닝] pytorch로 딥러닝 할 때 중요한 부분 (Dataset) pytorch를 배우면서 가장 두려웠던 부분이 Dataset을 어떻게 넣어야 할까 였습니다. 그래서 이 부분을 설명하고자 포스팅합니다(솔직히 pytorch 공홈에 보면 다 나와있더군요). 파이토치 공홈에 보면 다음과 같이 나와있습니다. https://pytorch.org/tutorials/beginner/basics/data_tutorial.html Datasets & DataLoaders — PyTorch Tutorials 1.12.0+cu102 documentation Note Click here to download the full example code Learn the Basics || Quickstart || Tensors || Datasets & DataLoaders || Transforms.. 2022. 7. 27.
[딥러닝공부] Model의 기본 - pytorch 저번 포스팅에 이어서 pytorch에서는 model을 어떻게 제작을 하는지 공유드리고자 합니다. 포스팅 내용이 괜찮았다면 주변 광고 클릭 몇번 눌러주시면 감사하겠습니다. pytorch는 tensorflow와 같이 python에서 Deep learning할때 필수적으로 사용되는 양대산맥이라고 할 수가 있습니다. 뭐 들리는 소문으로는 Tensorflow보다는 Pytorch가 좀 더 최근 트렌드에서 선호된다는 말을 들었던것 같습니다(그러나 모델 제작 정보는 tensorflow가 아직까지는 더 많은 것 같습니다.). pytorch의 경우 기본적인 model의 틀이 tensorflow의 GradientTape를 사용하는 방법과 유사하다고 생각될 수 있습니다(GradientTape와 같이 loss들을 업데이트 해주.. 2022. 5. 17.
[pytorch 설정] pytorch gpu 설정하는 방법 최근들어 Deep learning 관련 연구를 진행하게 되었습니다. deep learning을 진행하기 위해서는 R 보다는 python에서 실행하는것이 더 효율적입니다. (더 빠르고 정보가 많습니다.) 예전에 깔짝되던 Tensorflow, pytorch 중에 고민을 하다가 pytorch로 분석을 시작하게 되었고, pytorch중에서 GPU를 사용하는 방법에 대해서 설명하겠습니다. 먼저 cuda설정이 되어있는지 확인하려면 다음과 같이 python 코드를 짜면 됩니다. import torch print(torch.cuda.is_available()) # False = cpu # True = gpu 사용가능 cuda가 설정이 안되어 있으면 아래 쭈욱 읽어보시면 됩니다. 스펙은 다음과 같습니다. (그냥 컴퓨존에.. 2021. 12. 3.
[pytorch] tensor를 이해해보자! tensor라 함은 3차원 이상의 데이터를 이야기합니다. 0차원은 scaler, 1차원은 vector, 2차원은 matrix (행렬) 이라고 하죠. 우리가 딥러닝을 배우면서 많이 다루게될 데이터는 2차원 이상의 데이터입니다. 그렇다면 2차원 이상의 데이터를 어떻게 생각해야할까요?! 아래 그림을 보면서 설명하도록 하겠습니다. 위 사람사진에 대해서 얻을수 있는 데이터를 상상해 봅시다. 유전자에 대해서 알수가 있을 것 같고, 뼈사진을 통해 건강한 뼈를 가진 사람인지도 알 수가 있을것같구요. 그리고 키도 알 수가 있겠죠. (정확한 수치는 사진에 나오지 않았지만 알수있다고 가정해봅시다.) 한 사람에서 나오는 데이터의 양을 유전자, 뼈상태, 키 이렇게 대략 알 수가 있을 것입니다. 여기서 유전자, 뼈상태, 키 각각.. 2021. 8. 5.