Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

scale_img() bug fix

This commit is contained in:
Glenn Jocher 2020-04-07 12:59:52 -07:00
parent 68f58f4dec
commit 067ee264c0
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
utils/torch_utils.py
View File

@ -120,11 +120,10 @@ def scale_img(img, ratio=1.0, same_shape=True): # img(16,3,256,416), r=ratio
h, w = img.shape[2:] h, w = img.shape[2:]
s = (int(h * ratio), int(w * ratio)) # new size s = (int(h * ratio), int(w * ratio)) # new size
img = F.interpolate(img, size=s, mode='bilinear', align_corners=False) # resize img = F.interpolate(img, size=s, mode='bilinear', align_corners=False) # resize
if not same_shape: # pad/crop img
p = h - s[0], w - s[1] p = h - s[0], w - s[1]
if not same_shape: # pad/crop img
p = [int(math.fmod(x, 64)) for x in p] # pad to 64 grid size p = [int(math.fmod(x, 64)) for x in p] # pad to 64 grid size
img = F.pad(img, [0, p[1], 0, p[0]], value=0.447) # value = imagenet mean return F.pad(img, [0, p[1], 0, p[0]], value=0.447) # value = imagenet mean
return img
class ModelEMA: class ModelEMA: