This commit is contained in:
Glenn Jocher 2018-08-26 19:40:30 +02:00
parent af7144ba79
commit 56badeef8a
2 changed files with 3 additions and 3 deletions

View File

@ -1,6 +1,6 @@
classes=80 classes=80
train=/Users/glennjocher/Downloads/DATA/coco/trainvalno5k.part train=/Users/glennjocher/Downloads/DATA/coco/trainvalno5k.txt
valid=/Users/glennjocher/Downloads/DATA/coco/5k.part valid=/Users/glennjocher/Downloads/DATA/coco/5k.txt
names=data/coco.names names=data/coco.names
backup=backup/ backup=backup/
eval=coco eval=coco

View File

@ -36,7 +36,7 @@ def main(opt):
if platform == 'darwin': # macos if platform == 'darwin': # macos
train_path = data_config['valid'] train_path = data_config['valid']
else: # linux (gcp cloud) else: # linux (gcp cloud)
train_path = '../coco/trainvalno5k.part' train_path = '../coco/trainvalno5k.txt'
# Initialize model # Initialize model
model = Darknet(opt.cfg, opt.img_size) model = Darknet(opt.cfg, opt.img_size)