Commit Graph

629 Commits

Author SHA1 Message Date
Glenn Jocher c50df0d1db ONNX export compatability updates 2018-12-23 12:51:02 +01:00
Glenn Jocher 69963ff1f5 updates 2018-12-23 10:53:47 +01:00
Glenn Jocher aa3d1a2bbd updates 2018-12-22 14:25:02 +01:00
Glenn Jocher 465f847660 updates 2018-12-22 13:54:02 +01:00
Glenn Jocher de2d835b91 updates 2018-12-22 13:53:45 +01:00
Glenn Jocher 47df31a270 updates 2018-12-22 13:20:40 +01:00
Glenn Jocher 9c3d9dca97 updates 2018-12-22 13:09:05 +01:00
Glenn Jocher a21c131dd0 updates 2018-12-22 13:05:52 +01:00
Glenn Jocher ffd45ebf0c updates 2018-12-22 12:58:59 +01:00
Glenn Jocher a0ab4916fd yolov3-tiny addition 2018-12-22 12:49:55 +01:00
Glenn Jocher bd5b9693bf enable yolov3-tiny inference 2018-12-22 12:36:33 +01:00
Glenn Jocher 34fb1bb8a9 updates 2018-12-20 22:08:47 +01:00
Glenn Jocher 62c186da25 updates 2018-12-19 23:48:52 +01:00
Glenn Jocher b48c108ba0 updates 2018-12-17 22:43:55 +01:00
Glenn Jocher 682fd61385 updates 2018-12-17 22:43:30 +01:00
Glenn Jocher 89e8468895
Merge pull request #48 from jveitchmichaelis/patch-1
Remove auto-shutdown from get coco script
2018-12-17 21:49:55 +01:00
Josh Veitch-Michaelis 96d85ad4ba
Remove auto-shutdown from get coco script
This is presumably for unattended download on cloud systems, but the script should alert the user first. Automatically shutting down a system when you download some data shouldn't be default behaviour. It's also not in the original Darknet script (https://github.com/pjreddie/darknet/blob/master/scripts/get_coco_dataset.sh).

Alternatively run `get_coco_dataset.sh && sudo shutdown`.
2018-12-17 16:29:33 +00:00
Glenn Jocher bf23be9965 updates 2018-12-16 15:16:52 +01:00
Glenn Jocher 18ccd184bf updates 2018-12-16 15:16:19 +01:00
Glenn Jocher b52a49cf12 updates 2018-12-15 21:06:39 +01:00
Glenn Jocher b079c1b10c updates 2018-12-15 21:01:14 +01:00
Glenn Jocher 900851200e updates 2018-12-15 20:52:35 +01:00
Glenn Jocher 21ab0c76fd updates 2018-12-12 17:27:52 +01:00
Glenn Jocher 3c95b5c104 updates 2018-12-12 17:26:46 +01:00
Glenn Jocher b5a2747a6a updates 2018-12-12 17:02:37 +01:00
Glenn Jocher c591936446 updates 2018-12-11 21:49:56 +01:00
Glenn Jocher 7fb729269b Merge remote-tracking branch 'origin/master' 2018-12-11 20:47:21 +01:00
Glenn Jocher e28ac3de29 updates 2018-12-11 20:46:46 +01:00
Glenn Jocher cb21b75920
Update README.md 2018-12-11 20:23:27 +01:00
Glenn Jocher 6c1cd4f3a2
Update README.md 2018-12-11 20:18:05 +01:00
Glenn Jocher 4f80ef3464
Update README.md 2018-12-11 19:54:31 +01:00
Glenn Jocher c8bd1778f2
Delete coco_training_loss.png 2018-12-11 19:46:39 +01:00
Glenn Jocher 2e5c72321f
Update README.md 2018-12-11 19:46:15 +01:00
Glenn Jocher 3fe3951268 updates 2018-12-10 13:19:13 +01:00
Glenn Jocher 362b41436a
Merge pull request #45 from guigarfr/argparse
Argparse PR
2018-12-10 12:47:31 +01:00
Glenn Jocher c63e96bc82
Merge branch 'master' into argparse 2018-12-06 13:13:35 +01:00
Glenn Jocher 27849f2474 updates 2018-12-06 13:01:49 +01:00
Guillermo García d03ce45da5 train.py freeze-darknet53 shortened to freeze and action store_true
Traing with freeze: python train.py --freeze
Train without freeze: python train.py

Note: in the actual version freeze is only for first epoche
2018-12-05 16:57:16 +01:00
Guillermo García 868a116750 train.py remove hardcoded weights/ path for weights.
If I want to store my weights in 'weights2' path:
python train.py --weights-path weights2

Default is the same: weights
2018-12-05 16:57:16 +01:00
Guillermo García 9c0c1f23ab scripts: use data config defined class names
Shorten name of --data-config-path argument to --data-config
2018-12-05 16:57:16 +01:00
Guillermo García 89daa407e5 train.py report argument as store_true
Default is false: python train.py
If want the report: python train.py --report
2018-12-05 16:57:16 +01:00
Guillermo García b1fb6fa33d train.py resume argument as store_true
Default is false.

If want to resume, call train.py --resume
2018-12-05 16:57:16 +01:00
Guillermo García c807c16b79 Fix argument parser bad practice
Keep parsing inside __main__ block and call methods with arguments

Add double -- for long argument names (- reserved for shortcuts)
2018-12-05 16:57:16 +01:00
Guillermo García 5a566454f5 Extract seed and cuda initialization utils 2018-12-05 11:55:27 +01:00
Glenn Jocher 45ee668fd7 updates 2018-12-04 19:20:09 +01:00
Glenn Jocher be8603b2dd updates 2018-12-04 19:17:03 +01:00
Glenn Jocher fd6619d773 updates 2018-12-03 22:33:25 +01:00
Glenn Jocher 10cca39934 updates 2018-12-03 21:08:45 +01:00
Glenn Jocher dc704edf17 updates 2018-12-03 20:56:54 +01:00
Glenn Jocher 43d74fd840 updates 2018-12-03 15:42:10 +01:00