diff --git a/detect.py b/detect.py index 985bb6a5..f6170071 100644 --- a/detect.py +++ b/detect.py @@ -1,7 +1,7 @@ -import shutil -from pathlib import Path import argparse +import shutil import time +from pathlib import Path from sys import platform from models import * diff --git a/models.py b/models.py index e83cfdcd..a8d5e5db 100755 --- a/models.py +++ b/models.py @@ -1,6 +1,6 @@ +import os from collections import defaultdict -import os import torch.nn as nn from utils.parse_config import * diff --git a/train.py b/train.py index 37f6b24d..5dbc9d77 100644 --- a/train.py +++ b/train.py @@ -1,13 +1,11 @@ import argparse import time +import test # Import test.py to get mAP after each epoch from models import * from utils.datasets import * from utils.utils import * -# Import test.py to get mAP after each epoch -import test - def train( cfg,