updates
This commit is contained in:
parent
119d39599e
commit
184db1fb10
|
@ -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
|
||||||
|
|
2
train.py
2
train.py
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue