YAD2K icon indicating copy to clipboard operation
YAD2K copied to clipboard

why input_image_shape = K.placeholder(shape=(2, )) in retrain.py?

Open jianing-sun opened this issue 6 years ago • 1 comments

Hi I'm trying to retrain a yolo model based on UNIMIB2016. I think the input image shape should be the model size which in this case should be (416., 416.). Why in the script it is (2,)? Another question, I run the code to do training, but the loss in validation set is always around 4000 and even if the loss goes down in the training set, it does have any improvement in the validation set. Does anyone know why is that?

Thanks!

jianing-sun avatar May 20 '18 16:05 jianing-sun

Hi, for the one, the input image shape is the dimension of (416, 416), so it is (2, ). Actually, it should be the original size of the image, so it may not be (416, 416), maybe (768, 516) for example.

sun-su avatar Sep 13 '18 02:09 sun-su