RuntimeError: output with shape[1,28,28] doesn't match the broadcast shape [3,28,28]
5531 단어 인공지능
RuntimeError: output with shape [1, 28, 28] doesn’t match the broadcast shape [3, 28, 28]
Pytroch를 달리는 MNIST 핸드폰 식별 예시에서 이 오류가 발생했습니다. 오류 위치 코드는 다음과 같습니다.transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])])
분석 오류는 미니스트가 그레이스케일 이미지이기 때문에 하나의 채널만 있고 귀일화할 때의 세 채널과 일치하지 않기 때문에 세 가지 해결 방법이 있다
채널 수 변경
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Lambda(lambda x: x.repeat(3,1,1)),
transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))
])
귀일화 매개 변수 변경 (1)
transform = transforms.Compose([
transforms.ToTensor()
transforms.Normalize(mean=[0.5], std=[0.5])
])
귀일화 매개 변수 변경 (2)
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.1307,), (0.3081,))
])
이 방법을 거울로 삼은 것은 VictorGui의 이 박문.및 40.1307
균일치와 표준차로 나뉘는데, 이 두 가지는 서로 다른 데이터 집합에 따라 계산 변화를 해야 한다.
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
SPRESENSE x Neural Network Console을 만져 보자 제1회 ~SPRESENS의 소개와 전체의 구성~
이 기사는 20일째 기사입니다.
전회는 Katsuaki Takagi씨의 「 」이었습니다.
Sony가 발매한 보드 「SPRESENSE」와 「Neural Network Console」을 맞추어 사용해 봅니다.
연재 기사...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])])
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Lambda(lambda x: x.repeat(3,1,1)),
transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))
])
transform = transforms.Compose([
transforms.ToTensor()
transforms.Normalize(mean=[0.5], std=[0.5])
])
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.1307,), (0.3081,))
])
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
SPRESENSE x Neural Network Console을 만져 보자 제1회 ~SPRESENS의 소개와 전체의 구성~이 기사는 20일째 기사입니다. 전회는 Katsuaki Takagi씨의 「 」이었습니다. Sony가 발매한 보드 「SPRESENSE」와 「Neural Network Console」을 맞추어 사용해 봅니다. 연재 기사...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.