홈>2020-02-26 00:30

python - 저장된 CNN에서 교육 이미지에 할당 된 클래스 확률을 추출 할 수 있습니까?

분류 작업을 수행하도록 CNN을 훈련 시켰으며 vgg16 를 사용했습니다. categorical_crossentropy 와 함께 손실 함수, sgd 옵티 마이저 및 softmax 활성화 기능.

교육 된 모델을 h5 파일에 저장했습니다. 보이지 않는 새로운 이미지에 대한 클래스 확률 벡터를 얻는 함수는 predict 입니다. .

<올>

그러나 내가 알고 싶은 것은 훈련 이미지에 할당 된 최종 클래스 확률이 모델과 함께 저장되는지 여부입니다.

있는 경우 predict 를 사용하지 않고 저장된 모델에서이 정보를 추출하는 방법 기능?

답변 # 1
와이즈 비즈 와이즈 비즈 큰 저장에 적합한 파일 형식입니다 다차원 숫자 형 배열 모음 (예 : 모델, 데이터) 파일).
<올>
예 훈련 중"교육 이미지에 할당 된 최종 클래스 확률"을

에 저장할 수 있습니다. 파일.

하지만"만 훈련 된 모델을 HDF5 (.h5, .hdf5) 에 저장 한 이후 파일 "이 파일에는 네트워크 모델 (무게, 아키텍처 및 최적화 구성이 정의 된 경우) 만이 파일에 있습니다 (자세한 내용은 여기-"HDF5 파일 "섹션 참조).

여기서 가장 쉬운 옵션은 이미 언급 한 것입니다 :"예측 기능을 사용할 필요성이있는[out]으로 저장된 모델에서이 정보를 추출하십시오.
.h5