glenn-jocher commited on
Commit
4e62eb9
1 Parent(s): 2b7a113

check_img_size() int() update

Browse files
Files changed (1) hide show
  1. utils/utils.py +1 -1
utils/utils.py CHANGED
@@ -47,7 +47,7 @@ def check_git_status():
47
 
48
  def check_img_size(img_size, s=32):
49
  # Verify img_size is a multiple of stride s
50
- new_size = make_divisible(img_size, s) # ceil gs-multiple
51
  if new_size != img_size:
52
  print('WARNING: --img-size %g must be multiple of max stride %g, updating to %g' % (img_size, s, new_size))
53
  return new_size
 
47
 
48
  def check_img_size(img_size, s=32):
49
  # Verify img_size is a multiple of stride s
50
+ new_size = make_divisible(img_size, int(s)) # ceil gs-multiple
51
  if new_size != img_size:
52
  print('WARNING: --img-size %g must be multiple of max stride %g, updating to %g' % (img_size, s, new_size))
53
  return new_size