Spaces:
Build error
Build error
Update yolo.py
Browse files
yolo.py
CHANGED
|
@@ -18,7 +18,7 @@ from general import LOGGER, check_version, check_yaml, make_divisible, print_arg
|
|
| 18 |
from plots import feature_visualization
|
| 19 |
from torch_utils import (fuse_conv_and_bn, initialize_weights, model_info, profile, scale_img, select_device,
|
| 20 |
time_sync)
|
| 21 |
-
from
|
| 22 |
|
| 23 |
try:
|
| 24 |
import thop # for FLOPs computation
|
|
|
|
| 18 |
from plots import feature_visualization
|
| 19 |
from torch_utils import (fuse_conv_and_bn, initialize_weights, model_info, profile, scale_img, select_device,
|
| 20 |
time_sync)
|
| 21 |
+
from anchor_generator import make_anchors, dist2bbox
|
| 22 |
|
| 23 |
try:
|
| 24 |
import thop # for FLOPs computation
|