tflearn mismatch tensor shape

Question

이상한 오류. 나는 컬러 이미지를 사용하고 있었고 모든 것이 잘 작동했다. 이제는 시간과 속도 프로세스를 절약하기 위해 한 차원의 회색 이미지로 작업하기로 결정했습니다 (4 이전). 이 코드는 몇 가지 변경 사항과 함께 이전과 거의 동일합니다. 하지만 이제 모양이 같지 않다는 오류가 발생합니다. 어떤 도움이 필요합니까?

X, Y = image_preloader (dataset_file, image_shape = (704,800), MODE = "파일"categoriacal_labels = TRUE, = TRUE, files_extension = "JPEG"gray_scale 정상화 = TRUE)

네트워크 = input_data (모양 = (없음, 704, 800, 1)

ValueError : Tensor 'InputData/X : 0'에 모양 (',?, 704, 800)의 모양 값 , 1) '

Answer 1

나는 이것이 조금 오래되었음을 알고 있지만 잘하면 그것은 당신이나 다른 사람을 돕는다.

귀하의 그레이 스케일 이미지는 더 이상 RGB 데이터/채널에 여러 차원이 없다, 그래서 당신은 그것을 변경하여 input_data 형상 매개 변수에이를 반영해야한다 :

network = input_data(shape=[None, 704, 800])