코드 가져와봐 pretrained 그대로 가져다쓰고 마지막에 classifier 부분 안 박은거 아냐?
맞음 classifier 안쓰고 특징 추출만 하려고 로딩할때 models.vgg16(pretrained=True).features로 가져옴
그럼 feature가 몇차원으로 나오던가 상관이 없잖아
이미지 크기랑 레이어 해놓은대로 나오겠지.
코드 가져와봐 pretrained 그대로 가져다쓰고 마지막에 classifier 부분 안 박은거 아냐?
맞음 classifier 안쓰고 특징 추출만 하려고 로딩할때 models.vgg16(pretrained=True).features로 가져옴
그럼 feature가 몇차원으로 나오던가 상관이 없잖아
이미지 크기랑 레이어 해놓은대로 나오겠지.