updates
This commit is contained in:
parent
c99cb43c55
commit
345f4773b7
2
test.py
2
test.py
|
@ -28,7 +28,7 @@ num_classes = int(data_config['classes'])
|
||||||
if platform == 'darwin': # MacOS (local)
|
if platform == 'darwin': # MacOS (local)
|
||||||
test_path = data_config['valid']
|
test_path = data_config['valid']
|
||||||
else: # linux (cloud, i.e. gcp)
|
else: # linux (cloud, i.e. gcp)
|
||||||
test_path = '../coco/trainvalno5k.part'
|
test_path = '../coco/5k.part'
|
||||||
|
|
||||||
# Initiate model
|
# Initiate model
|
||||||
model = Darknet(opt.cfg, opt.img_size)
|
model = Darknet(opt.cfg, opt.img_size)
|
||||||
|
|
Loading…
Reference in New Issue