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

updates

This commit is contained in:
Glenn Jocher 2018-09-01 18:41:05 +02:00
parent 382100307a
commit 8fd8d8eb04
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -19,7 +19,7 @@ Python 3.6 or later with the following `pip3 install -U -r requirements.txt` pac
# Training # Training
Run `train.py` to begin training after downloading COCO data with `data/get_coco_dataset.sh`. Run `train.py` to begin training after downloading COCO data with `data/get_coco_dataset.sh` and specifying COCO path on line 37 (local) or line 39 (cloud).
Run `train.py -resume 1` to resume training from the most recently saved checkpoint `checkpoints/latest.pt`. Run `train.py -resume 1` to resume training from the most recently saved checkpoint `checkpoints/latest.pt`.

4
train.py
View File

@ -33,9 +33,9 @@ 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)
num_classes = int(data_config['classes']) num_classes = int(data_config['classes'])
if platform == 'darwin': # macos if platform == 'darwin': # MacOS (local)
train_path = data_config['valid'] train_path = data_config['valid']
else: # linux (gcp cloud) else: # linux (cloud, i.e. gcp)
train_path = '../coco/trainvalno5k.part' train_path = '../coco/trainvalno5k.part'
# Initialize model # Initialize model