把预测文件predict_add.py,从YOLOv3_TensorFlow-master文件夹中移动到根目录下,如果直接执行会出现以下错误。
wu@wu-X555LF:~$ python predict_add.py
Traceback (most recent call last):
File "predict_add.py", line 11, in <module>
from utils.misc_utils import parse_anchors, read_class_names
ImportError: No module named utils.misc_utils
wu@wu-X555LF:~$ python predict_add.py
Traceback (most recent call last):
File "predict_add.py", line 16, in <module>
from model import yolov3
ImportError: No module named model
看部分引入的包头,除了tensorflow,numpy,argparse,cv2,os是系统安装的包; utils,model是本地目录加载的包,Python首先会在系统目录下查找所有的包,如果没有查找到指定的包,会在当前目录查找,最后还是找不到,会报错。所以需要把utils,model包,放在同一目录下。
# coding: utf-8
from __future__ import division, print_function
import tensorflow as tf
import numpy as np
import argparse
import cv2
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
from utils.misc_utils import parse_anchors, read_class_names
from utils.nms_utils import gpu_nms
from utils.plot_utils import get_color_table, plot_one_box
from utils.data_aug import letterbox_resize
from model import yolov3