diff --git a/models/export.py b/models/export.py index 033b901..06d51f2 100644 --- a/models/export.py +++ b/models/export.py @@ -8,8 +8,8 @@ import argparse import torch -from utils.google_utils import attempt_download from utils.general import set_logging +from utils.google_utils import attempt_download if __name__ == '__main__': parser = argparse.ArgumentParser() diff --git a/models/yolo.py b/models/yolo.py index 573a3ab..3aaa422 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -1,6 +1,6 @@ import argparse -import math import logging +import math from copy import deepcopy from pathlib import Path diff --git a/utils/general.py b/utils/general.py index ede29f0..ae30864 100755 --- a/utils/general.py +++ b/utils/general.py @@ -1275,3 +1275,4 @@ def plot_results(start=0, stop=0, bucket='', id=(), labels=(), fig.tight_layout() ax[1].legend() fig.savefig(Path(save_dir) / 'results.png', dpi=200) + diff --git a/utils/google_utils.py b/utils/google_utils.py index 5740ece..5c60fa2 100644 --- a/utils/google_utils.py +++ b/utils/google_utils.py @@ -6,6 +6,7 @@ import os import platform import time from pathlib import Path + import torch diff --git a/utils/torch_utils.py b/utils/torch_utils.py index a5d898a..c024983 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -1,7 +1,7 @@ +import logging import math import os import time -import logging from copy import deepcopy import torch