Commit Graph

249 Commits

Author SHA1 Message Date
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
Glenn Jocher b64620cf75 updates 2018-12-03 14:12:46 +01:00
Glenn Jocher 40b536a426 updates 2018-12-03 14:08:59 +01:00
Glenn Jocher 147dfe10d8 Merge remote-tracking branch 'origin/master' 2018-12-03 14:05:57 +01:00
Glenn Jocher 5843c41dfc add multi_scale support 2018-12-03 14:05:50 +01:00
Glenn Jocher 4edd41e2e8
Update README.md 2018-12-03 14:03:27 +01:00
Glenn Jocher f05934f2eb updates 2018-12-03 01:36:03 +01:00
Glenn Jocher 448a8f0f4b updates 2018-12-01 12:09:40 +01:00
Glenn Jocher b0c0182062 updates 2018-11-30 11:57:14 +01:00
Glenn Jocher 0240ac44f6 updates 2018-11-30 11:56:38 +01:00
Glenn Jocher 35e445c5da updates 2018-11-29 22:10:35 +01:00
Glenn Jocher bd649f241f updates 2018-11-29 12:12:48 +01:00
Glenn Jocher af0033c9e9 updates 2018-11-29 11:59:29 +01:00
Glenn Jocher d5331be0a0 updates 2018-11-29 11:43:19 +01:00
Glenn Jocher 053566b174 updates 2018-11-28 10:27:55 +01:00
Glenn Jocher cc419d88ea updates 2018-11-28 10:25:00 +01:00
Glenn Jocher 5a0575af3a updates 2018-11-27 18:43:46 +01:00
Glenn Jocher b07ee41867 updates 2018-11-27 18:14:48 +01:00
Glenn Jocher ab9ee6aa9a updates 2018-11-23 19:45:39 +01:00
Glenn Jocher 82124805f8 updates 2018-11-23 18:13:35 +01:00
Glenn Jocher 7b13af707d updates 2018-11-23 18:09:47 +01:00
Glenn Jocher bf66656b4e updates 2018-11-23 15:34:49 +01:00
Glenn Jocher 6e825acb72 updates 2018-11-23 15:32:41 +01:00
Glenn Jocher 887ab29c64 updates 2018-11-22 20:03:09 +01:00
Glenn Jocher 6f83c321c8 updates 2018-11-22 20:02:11 +01:00
Glenn Jocher 7ca924b172 updates 2018-11-22 17:17:07 +01:00
Glenn Jocher 075b629049 updates 2018-11-22 17:16:17 +01:00
Glenn Jocher 120af70798 updates 2018-11-22 17:13:47 +01:00
Glenn Jocher 57f2b3f6d7 updates 2018-11-22 16:42:58 +01:00
Glenn Jocher 06579775a3
Merge pull request #37 from nirbenz/nirbenz
Fixed NMS bug causing big CPU usage.
2018-11-22 16:29:28 +01:00
Glenn Jocher 154fae4430 updates 2018-11-22 15:04:02 +01:00