Optimize imports (#9464)

segment/train.py

@@ -39,8 +39,6 @@ if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
 ROOT = Path(os.path.relpath(ROOT, Path.cwd()))  # relative
 
-import torch.nn.functional as F
-
 import segment.val as validate  # for end-of-epoch mAP
 from models.experimental import attempt_load
 from models.yolo import SegmentationModel

utils/loggers/clearml/clearml_utils.py

@@ -11,6 +11,7 @@ from utils.plots import Annotator, colors
 try:
     import clearml
     from clearml import Dataset, Task
+
     assert hasattr(clearml, '__version__')  # verify package import not local dir
 except (ImportError, AssertionError):
     clearml = None

utils/loggers/comet/hpo.py

@@ -14,7 +14,7 @@ ROOT = FILE.parents[3]  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
 
-from train import parse_opt, train
+from train import train
 from utils.callbacks import Callbacks
 from utils.general import increment_path
 from utils.torch_utils import select_device