add stride order reversal for c53*.cfg

This commit is contained in:
Glenn Jocher 2020-05-17 22:45:48 -07:00
parent bc9da228e0
commit eacded6a2c
1 changed files with 2 additions and 0 deletions

View File

@ -848,6 +848,8 @@ def plot_wh_methods(): # from utils.utils import *; plot_wh_methods()
def plot_images(images, targets, paths=None, fname='images.jpg', names=None, max_size=640, max_subplots=16):
tl = 3 # line thickness
tf = max(tl - 1, 1) # font thickness
if os.path.isfile(fname): # do not overwrite
return None
if isinstance(images, torch.Tensor):
images = images.cpu().numpy()