optimize imports
This commit is contained in:
parent
7d5878872c
commit
9706002b71
|
@ -1,7 +1,7 @@
|
||||||
import shutil
|
|
||||||
from pathlib import Path
|
|
||||||
import argparse
|
import argparse
|
||||||
|
import shutil
|
||||||
import time
|
import time
|
||||||
|
from pathlib import Path
|
||||||
from sys import platform
|
from sys import platform
|
||||||
|
|
||||||
from models import *
|
from models import *
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
|
import os
|
||||||
from collections import defaultdict
|
from collections import defaultdict
|
||||||
|
|
||||||
import os
|
|
||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
|
|
||||||
from utils.parse_config import *
|
from utils.parse_config import *
|
||||||
|
|
4
train.py
4
train.py
|
@ -1,13 +1,11 @@
|
||||||
import argparse
|
import argparse
|
||||||
import time
|
import time
|
||||||
|
|
||||||
|
import test # Import test.py to get mAP after each epoch
|
||||||
from models import *
|
from models import *
|
||||||
from utils.datasets import *
|
from utils.datasets import *
|
||||||
from utils.utils import *
|
from utils.utils import *
|
||||||
|
|
||||||
# Import test.py to get mAP after each epoch
|
|
||||||
import test
|
|
||||||
|
|
||||||
|
|
||||||
def train(
|
def train(
|
||||||
cfg,
|
cfg,
|
||||||
|
|
Loading…
Reference in New Issue