caffe digits 환경에서finetune 시 prototxt 파일 수정 방법
data 층에 있는 여분의 매개 변수를 삭제하고 나머지는 다음과 같다.layer {
name: "data"
type: "Data"
top: "data"
top: "label"
include {
phase: TRAIN
}
transform_param {
mirror: true
crop_size: 227
}
}layer {
name: "data"
type: "Data"
top: "data"
top: "label"
include {
phase: TEST
}
transform_param {
mirror: false
crop_size: 227
}
}2、모든
include {phase: TRAIN}을 include {stage: "train"}으로 교체하고, 모든 include {phase: TEST}을 include {stage: "val"}으로 교체하여loss층에 exclude {stage: "deploy"}을 추가3、fc8층의 이름을
fc8-re(이름만 바꾸면 됩니다. 분류 수량과pretrained 모델 수량이 동시에 초기화되지 않기 때문에) fc8의 num_output 줄을 삭제합니다.4. 파일 마지막에 softmax 층 추가
layer {
name: "prob"
type: "Softmax"
bottom: "fc8"
top: "prob"
include{
stage:"deploy"
}
}이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
다양한 언어의 JSONJSON은 Javascript 표기법을 사용하여 데이터 구조를 레이아웃하는 데이터 형식입니다. 그러나 Javascript가 코드에서 이러한 구조를 나타낼 수 있는 유일한 언어는 아닙니다. 저는 일반적으로 '객체'{}...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.
좋은 웹페이지 즐겨찾기
