2016-07-03 4 views
0

토치가있는 훈련 된 VGG를 통해 내 시스템에서 기존 이미지를 전달할 수있는 방법이 있습니까? 우분투 14.04를 사용하고 있으며 불행히도 GPU가 없습니다. 꽤 광범위하게 검색했지만 찾은 모든 것들은 GPU가 필요합니다. 토치없이 VGG를 사용하는 다른 방법이 있습니까? 제안을 할 수는 있지만 GPU가 필요하지는 않습니다.토치 훈련 된 VGG 얼굴 사용

답변

0

GPU를 사용하여 네트워크를 실행하면 작업 속도가 빨라지지만 CPU 모드에서만 네트워크를 실행할 수 있습니다. 당신이 모델을로드하면

는 GPU에 pretrained 할 수 있습니다 다음과 같은 CPU 간단한 변환을 : 당신은 쉽게 image 라이브러리의 도움으로 컴퓨터에서 이미지를로드 한 후 할 수있는

model = model:float() 

전달 패스

local img = image.load(imagefile,3,'byte') local output = model:forward(img)

+0

고마워요! 어떤 VGG 모델에 대해 이야기하고 있으며 어디에서 찾을 수 있습니까? – Sibi

+0

@Sibi https://github.com/torch/torch7/wiki/ModelZoo (이들은 caffe 모델 임) 사용 가능한 사전 네트워크 목록을 찾을 수 있으며 loadcaffe 모듈을 사용하여 토치 모델로로드 할 수 있습니다. 위에서 설명한 링크에서 설명합니다. –

관련 문제