add stride order reversal for c53*.cfg
This commit is contained in:
parent
bc9da228e0
commit
eacded6a2c
|
@ -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):
|
def plot_images(images, targets, paths=None, fname='images.jpg', names=None, max_size=640, max_subplots=16):
|
||||||
tl = 3 # line thickness
|
tl = 3 # line thickness
|
||||||
tf = max(tl - 1, 1) # font thickness
|
tf = max(tl - 1, 1) # font thickness
|
||||||
|
if os.path.isfile(fname): # do not overwrite
|
||||||
|
return None
|
||||||
|
|
||||||
if isinstance(images, torch.Tensor):
|
if isinstance(images, torch.Tensor):
|
||||||
images = images.cpu().numpy()
|
images = images.cpu().numpy()
|
||||||
|
|
Loading…
Reference in New Issue