img_size

[nhathoang0110]

How do I choose img_size for the model? Thanks you

[ZhengPeng7]

Search the variable keep_res in dataset.py and you will know how to do it.

[nhathoang0110]

keep_res = False #
if keep_res:
input_h = (height | 31) + 1
input_w = (width | 31) + 1
s = np.array([input_w, input_h], dtype=np.float32)
else:
s = max(img.shape[0], img.shape[1]) * 1.0
input_h, input_w = 512, 512

Did I just change input_h and input_w in the last command line?
Thanks very much