Vvaann commited on
Commit
e94074c
·
verified ·
1 Parent(s): 40f326d

Update yolo.py

Browse files
Files changed (1) hide show
  1. yolo.py +1 -1
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 tal.anchor_generator import make_anchors, dist2bbox
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