python yolov5的封装

• 目录结构
  
  DRImgDetect.py 调用DRDetect.y，即调用yolov5的检测函数

• DRImgDetect.py

import time
import cv2
import torch
import yaml
import torch.backends.cudnn as cudnn
from numpy import random
import numpy as np
import os
import sys

from DRDetect.DRDetect import DRDetection


if __name__ == '__main__':

    imgpath=''
    modelPath = " "
    DRDet = DRDetection(path = modelPath)
    objects = DRDet.predict(imgpath)
    print(objects)

• 重点

需加入sys.path.insert(0,‘/path’)，这样新添加的目录会优先于其他目录被import检查,不然会报错