-
目录结构
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检查,不然会报错