updates
This commit is contained in:
parent
be8603b2dd
commit
45ee668fd7
3
test.py
3
test.py
|
@ -26,10 +26,7 @@ def main(opt):
|
|||
# Configure run
|
||||
data_config = parse_data_config(opt.data_config_path)
|
||||
nC = int(data_config['classes']) # number of classes (80 for COCO)
|
||||
if platform == 'darwin': # MacOS (local)
|
||||
test_path = data_config['valid']
|
||||
else: # linux (cloud, i.e. gcp)
|
||||
test_path = '../coco/5k.part'
|
||||
|
||||
# Initiate model
|
||||
model = Darknet(opt.cfg, opt.img_size)
|
||||
|
|
Loading…
Reference in New Issue