Add check_requirements() (#1853)

* Add check_requirements()

* add import

* parameterize filename

* add to detect, test
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-01-06 16:35:40 -08:00
1 parent 135ec5c5ce
commit e77c77f580
4 files changed
+16 -5

No files matched your search

+2 -1
View File
@@ -28,7 +28,7 @@ from utils.autoanchor import check_anchors
from utils.datasets import create_dataloader
from utils.general import labels_to_class_weights, increment_path, labels_to_image_weights, init_seeds, \
fitness, strip_optimizer, get_latest_run, check_dataset, check_file, check_git_status, check_img_size, \
print_mutation, set_logging, one_cycle
check_requirements, print_mutation, set_logging, one_cycle
from utils.google_utils import attempt_download
from utils.loss import compute_loss
from utils.plots import plot_images, plot_labels, plot_results, plot_evolution
@@ -472,6 +472,7 @@ if __name__ == '__main__':
set_logging(opt.global_rank)
if opt.global_rank in [-1, 0]:
check_git_status()
check_requirements()
# Resume
if opt.resume: # resume an interrupted run