Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-02-09 19:13:07 +01:00
parent e88798aefd
commit a0936a4eac
2 changed files with 5 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
detect.py
View File

@ -96,6 +96,8 @@ def detect(cfg, weights, images, output='output', img_size=416, conf_thres=0.3,
if platform == 'darwin': # MacOS if platform == 'darwin': # MacOS
os.system('open ' + output) os.system('open ' + output)
os.system('open ' + save_img_path)
if __name__ == '__main__': if __name__ == '__main__':

25
test.py
View File

@ -7,16 +7,7 @@ from utils.utils import *
from utils import torch_utils from utils import torch_utils
def test( def test(cfg, data_cfg, weights, batch_size=16, img_size=416, iou_thres=0.5, conf_thres=0.3, nms_thres=0.45):
cfg,
data_cfg,
weights,
batch_size=16,
img_size=416,
iou_thres=0.5,
conf_thres=0.3,
nms_thres=0.45,
):
device = torch_utils.select_device() device = torch_utils.select_device()
# Configure run # Configure run
@ -135,15 +126,5 @@ if __name__ == '__main__':
opt = parser.parse_args() opt = parser.parse_args()
print(opt, end='\n\n') print(opt, end='\n\n')
init_seeds() mAP = test(opt.cfg, opt.data_cfg, opt.weights, opt.batch_size, opt.img_size, opt.iou_thres, opt.conf_thres,
opt.nms_thres)
mAP = test(
opt.cfg,
opt.data_cfg,
opt.weights,
batch_size=opt.batch_size,
img_size=opt.img_size,
iou_thres=opt.iou_thres,
conf_thres=opt.conf_thres,
nms_thres=opt.nms_thres,
)