From 728efa6576eae595ddcfdd8b75ab5da40ddfcaf4 Mon Sep 17 00:00:00 2001 From: lorenzomammana Date: Tue, 4 Aug 2020 21:22:15 +0200 Subject: [PATCH] Fix missing imports (#627) * Fix missing imports * Update detect.py Co-authored-by: Glenn Jocher --- detect.py | 3 ++- train.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/detect.py b/detect.py index ca7b96773173..95a17a157cab 100644 --- a/detect.py +++ b/detect.py @@ -12,7 +12,8 @@ from models.experimental import attempt_load from utils.datasets import LoadStreams, LoadImages -from utils.general import check_img_size, non_max_suppression, apply_classifier, scale_coords, xyxy2xywh, plot_one_box +from utils.general import ( + check_img_size, non_max_suppression, apply_classifier, scale_coords, xyxy2xywh, plot_one_box, strip_optimizer) from utils.torch_utils import select_device, load_classifier, time_synchronized diff --git a/train.py b/train.py index 915eef933056..8bf01e4c1bd5 100644 --- a/train.py +++ b/train.py @@ -24,7 +24,7 @@ from utils.general import ( check_img_size, torch_distributed_zero_first, labels_to_class_weights, plot_labels, check_anchors, labels_to_image_weights, compute_loss, plot_images, fitness, strip_optimizer, plot_results, - get_latest_run, check_git_status, check_file, increment_dir, print_mutation) + get_latest_run, check_git_status, check_file, increment_dir, print_mutation, plot_evolution) from utils.google_utils import attempt_download from utils.torch_utils import init_seeds, ModelEMA, select_device