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

updates

This commit is contained in:
Glenn Jocher 2018-12-04 19:20:09 +01:00
parent be8603b2dd
commit 45ee668fd7
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test.py
View File

@ -26,10 +26,7 @@ def main(opt):
# Configure run # Configure run
data_config = parse_data_config(opt.data_config_path) data_config = parse_data_config(opt.data_config_path)
nC = int(data_config['classes']) # number of classes (80 for COCO) nC = int(data_config['classes']) # number of classes (80 for COCO)
if platform == 'darwin': # MacOS (local) test_path = data_config['valid']
test_path = data_config['valid']
else: # linux (cloud, i.e. gcp)
test_path = '../coco/5k.part'
# Initiate model # Initiate model
model = Darknet(opt.cfg, opt.img_size) model = Darknet(opt.cfg, opt.img_size)