반응형
Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
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 |
Tags
- DeepLearning
- CNN
- objectdetection
- pychram
- AI
- 가상환경구축
- LSTM
- Deep learning
- torch.nn
- __call__
- vision transformer
- rnn
- Anaconda
- ubuntu
- 가상환경
- docker
- pip install
- pytorch
- torch.nn.Module
- 딥러닝
- __init__
- python 문법
- ViT
- tensorflow
- python
- 파이썬
- Torchvision
- wsl2
- 머신러닝
- 파이썬문법
Archives
- Today
- Total
목록ImageFolder (1)
인공지능을 좋아하는 곧미남
[pytorch] torchvision ImageFolder
오늘은 torchvision에서 제공하는 ImageFolder 클래스에 대해 알아보겠습니다.from torchvision.datasets import ImageFolder1. ImageFolder의 사용법1. ImageFolder의 사용법 - 하나의 폴더에 하위 폴더가 Class를 나누는 기준이 되어 Image Classification의 target 데이터를 얻을 때 정말 유용하다. 저는 이러한 tree 구조의 폴더로 구성된 데이터를 생성하여 테스트해보았습니다. 맨 마지막 하위 폴더에는 png 파일이 저장되어 있습니다. 즉, 상위 폴더 1개, 하위 폴더 여러개(Class 수 만큼), 맨 하위 폴더엔 각 Class별 이미지 파일이 저장되어 있습니다. 그 후 이렇게 ImageFolder은 세 가지 변..
code_study/pytorch
2022. 5. 16. 01:15