This commit is contained in:
Glenn Jocher 2019-04-09 12:24:32 +02:00
parent 26b115c306
commit 4b7eb0aec9
1 changed files with 1 additions and 0 deletions

View File

@ -488,6 +488,7 @@ def plot_wh_methods(): # from utils.utils import *; plot_wh_methods()
def plot_images(imgs, targets, fname='images.jpg'):
# Plots training images overlaid with targets
fig = plt.figure(figsize=(10, 10))
img_size = imgs.shape[3]
bs = imgs.shape[0] # batch size