분류 전체보기
-
[코드] Imagenet 분류 코드
프레임워크 : pytorch import torch import torchvision import torchvision.transforms as transforms 데이터 셋을 로드하는 코드 # Load the dataset transform = transforms.Compose([ transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) trainset = torchvision.datasets.ImageNet(root='./data', train=True, transform=transfo..
-
[Toy Project] YOLO를 이용한 Image Detection
Intro 많은 패션 사진으로 데이터셋을 만들기 위해 python의 모듈 Beautifulsoup과 selenium을 이용해서 이미지크롤링하여 DataSet을 만들고 YOLO 모델을 통해 Detection을 진행하는 프로젝트를 한다. Data 1. 데이터 https://www.musinsa.com/app/styles/lists 크롤링 여부를 확인하기 위해 /robots.txt를 사용해본다 Disallow라고 하니 하지 말라는 것이다. 아쉽지만 토이프로젝트로 많은 양의 데이터를 사용하지 않을 것이기 때문에 손수 다운로드하기로 했다. 약 400개의 이미지 파일을 만들었고 최대한 정면, 인식이 가능할 것 같은 사진 위주로 추렸다. Image Labeling 이미지 라벨링 작업을 진행하는데에는 labelImg ..