본문 바로가기

Debugging

InvalidArgumentError: Input to reshape is a tensor with 84500 values, but the requested shape requires a multiple of 5070 오류 해결

 

.cfg 파일을 수정해야 한다.

 

 

마지막 레이어의 filters 값을

filters = num * (#classes + 5)로 변경해주어야 한다.

 

여기서 num 값은

region 레이어의 num 값이며 한 cell 당 몇 개의 박스를 생성할지 결정한다.

 

본인의 경우 클래스 개수가 1개이므로 5*(1+5) = 30으로 설정했다.