Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity
2,473 Commits 1 Branch 0 Tags 19 MiB
Commit Graph

769 Commits

Author SHA1 Message Date
Glenn Jocher f527d30ccd updates 2019-04-02 18:50:55 +02:00
Glenn Jocher 7f220a14cb updates 2019-04-02 18:10:53 +02:00
Glenn Jocher 03559eff6e updates 2019-04-02 18:05:25 +02:00
Glenn Jocher 1457f66419 updates 2019-04-02 18:04:04 +02:00
Glenn Jocher d526ce0d11 updates 2019-04-02 16:33:52 +02:00
Glenn Jocher 3c233bc0b7 updates 2019-04-02 16:06:15 +02:00
Glenn Jocher 47400aa066 updates 2019-04-02 14:29:35 +02:00
Glenn Jocher af61da5d41 updates 2019-04-02 14:07:14 +02:00
Glenn Jocher 01569d15e3 updates 2019-04-02 13:43:18 +02:00
Glenn Jocher b56952d707 updates 2019-03-31 20:19:15 +02:00
Gabriel Bianconi 8901e96a38 Save model by default (#178) 
* Save model by default

* Update train.py
 2019-03-31 19:11:13 +02:00
Glenn Jocher c0cacc45a1
mAP Update (#176) 
* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates
 2019-03-30 18:45:04 +01:00
Glenn Jocher f5d398a68d updates 2019-03-25 15:33:36 +01:00
Glenn Jocher ff06b698ed updates 2019-03-25 15:32:00 +01:00
Glenn Jocher c16b587ac7
Update train.py 2019-03-25 15:06:22 +01:00
Glenn Jocher 06d264198c
Update train.py 2019-03-25 15:05:35 +01:00
Glenn Jocher c7192f64c9
Update train.py 2019-03-25 15:03:13 +01:00
Glenn Jocher cd51e1137b
Add collate_fn() to DataLoader (#163) 
Multi-GPU update with custom collate function to allow variable size target vector per image without needing to pad targets.
 2019-03-25 14:59:38 +01:00
Fatih Baltacı 9208a91095
Update train.py 2019-03-25 15:22:05 +03:00
perry0418 5daea5882f
Update train.py 
fix problem of multiple gpu training
 2019-03-25 16:13:21 +08:00
perry0418 386835d7ca
Update train.py 
solve the multi-gpu training problem.
 2019-03-25 14:56:38 +08:00
Glenn Jocher b31f8fb017 updates 2019-03-22 15:08:03 +02:00
Glenn Jocher 943db40f1a updates 2019-03-22 00:41:43 +02:00
Glenn Jocher 20beee0c5b updates 2019-03-21 22:51:22 +02:00
Glenn Jocher 176851f83a updates 2019-03-21 22:49:57 +02:00
Glenn Jocher 6aef4e6a78 updates 2019-03-21 22:41:12 +02:00
Glenn Jocher d047062074 updates 2019-03-21 17:26:31 +02:00
Glenn Jocher a3067e7978 multi_thread dataloader 2019-03-21 14:57:41 +02:00
Glenn Jocher 70fe2204b4 multi_thread dataloader 2019-03-21 14:48:40 +02:00
Glenn Jocher be38caf284 updates 2019-03-21 12:13:09 +02:00
Glenn Jocher 2856af5036 updates 2019-03-21 12:11:08 +02:00
Glenn Jocher aecf840701 updates 2019-03-21 12:09:54 +02:00
Glenn Jocher 03791babfb Merge branch 'master' of /Users/glennjocher/PycharmProjects/yolov3 with conflicts. 2019-03-21 12:08:55 +02:00
Glenn Jocher ad49e70f47
Merge pull request #145 from perry0418/master 
Update train.py
 2019-03-21 12:04:50 +02:00
Glenn Jocher d661fba8ae updates 2019-03-21 11:48:50 +02:00
perry0418 35396adc9c
Update train.py 
solve the Multi-GPU --resume Error #138
https://github.com/ultralytics/yolov3/issues/138
 2019-03-21 12:02:57 +08:00
Glenn Jocher 327aaebd7c updates 2019-03-20 22:10:18 +02:00
Glenn Jocher 735b1a370b multi_gpu multi_scale 2019-03-19 15:43:10 +02:00
Glenn Jocher dcdd1ae6b7 multi_gpu multi_scale 2019-03-19 15:35:12 +02:00
Glenn Jocher 76f555c108 multi_gpu multi_scale 2019-03-19 12:34:12 +02:00
Glenn Jocher 056eed2833 multi_gpu multi_scale 2019-03-19 10:38:32 +02:00
Glenn Jocher 45fac6bff1
multi_gpu (#135) 
* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates
 2019-03-17 23:45:39 +02:00
Glenn Jocher c1c09eb3cc
Update train.py 2019-03-10 15:03:17 +01:00
Glenn Jocher bc0f30933a updates 2019-03-07 17:16:38 +01:00
Glenn Jocher ff9d343019
Update train.py 2019-03-07 13:28:54 +01:00
Glenn Jocher 6fb14fc903 updates 2019-03-05 17:14:40 +01:00
Glenn Jocher a2ad00d6fc updates 2019-03-05 17:10:34 +01:00
Glenn Jocher e5dc942fee updates 2019-03-04 17:38:38 +01:00
Glenn Jocher 3bea4da604 updates 2019-03-04 17:34:53 +01:00
Glenn Jocher 65ed0d1122 updates 2019-03-04 15:46:25 +01:00
Glenn Jocher 90a20f93e5 updates 2019-02-26 02:53:11 +01:00
Glenn Jocher 9e60e97a6c updates 2019-02-22 16:24:30 +01:00
Glenn Jocher ac22a717f1 updates 2019-02-22 16:15:20 +01:00
Glenn Jocher 12e605165e updates 2019-02-22 15:05:03 +01:00
Glenn Jocher e62736f8a8 updates 2019-02-21 20:16:58 +01:00
Glenn Jocher 485321ecb1 updates 2019-02-21 16:18:11 +01:00
Glenn Jocher af853f604c updates 2019-02-21 15:57:18 +01:00
Glenn Jocher 0b971eddff updates 2019-02-20 18:41:31 +01:00
Glenn Jocher f8c675dbc0 updates 2019-02-20 17:44:41 +01:00
Glenn Jocher a65a383d3a updates 2019-02-20 13:00:39 +01:00
Glenn Jocher 15bba5a345 xy and wh losses respectively merged 2019-02-19 19:55:33 +01:00
Glenn Jocher c828f5459f select GPU0 if multiple available 2019-02-16 14:33:52 +01:00
Glenn Jocher 9706002b71 optimize imports 2019-02-12 18:05:58 +01:00
Glenn Jocher e4e64a9ff6 updates 2019-02-12 13:50:43 +01:00
Glenn Jocher 9f145d2aa7 updates 2019-02-11 22:44:25 +01:00
Glenn Jocher e23b1a3d73 webcam updates 2019-02-11 18:15:51 +01:00
Glenn Jocher 1ca352b328 class labeling corrections 2019-02-11 12:44:12 +01:00
Glenn Jocher ab2ea5a2f9 updates 2019-02-10 22:02:55 +01:00
Glenn Jocher 22dc8c0ea6 updates 2019-02-10 22:01:53 +01:00
Glenn Jocher 715c4575bf updates 2019-02-10 21:34:15 +01:00
Glenn Jocher 917f9dd248 updates 2019-02-10 21:28:27 +01:00
Glenn Jocher 6f0086103c updates 2019-02-10 21:10:50 +01:00
Glenn Jocher 1cd907c59b updates 2019-02-09 19:29:19 +01:00
Glenn Jocher be934ba5a5 updates 2019-02-09 19:26:53 +01:00
Glenn Jocher a701374014 updates 2019-02-09 19:24:51 +01:00
Glenn Jocher c2436d8197 updates 2019-02-08 22:43:05 +01:00
Glenn Jocher 8b88e50f2f updates 2019-02-08 16:50:48 +01:00
Glenn Jocher eec0dc7b6c ONNX compatibility updates 2018-12-28 20:09:06 +01: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 900851200e updates 2018-12-15 20:52:35 +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 3fe3951268 updates 2018-12-10 13:19:13 +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 fd6619d773 updates 2018-12-03 22:33:25 +01:00
Glenn Jocher 10cca39934 updates 2018-12-03 21:08:45 +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 5843c41dfc add multi_scale support 2018-12-03 14:05:50 +01:00
Glenn Jocher af0033c9e9 updates 2018-11-29 11:59:29 +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 887ab29c64 updates 2018-11-22 20:03:09 +01:00
Glenn Jocher 6f83c321c8 updates 2018-11-22 20:02:11 +01:00
Glenn Jocher b9d87be318 updates 2018-11-22 14:54:52 +01:00
Glenn Jocher db515a4535 updates 2018-11-22 14:14:19 +01:00
Glenn Jocher 809667404f updates 2018-11-22 13:52:22 +01:00
Glenn Jocher 4e4b67b3c5 updates 2018-11-21 18:01:18 +01:00
Glenn Jocher a021f97110 updates 2018-11-15 01:01:04 +01:00
Glenn Jocher 1ea87c49c4 updates 2018-11-15 00:57:15 +01:00
Glenn Jocher 45c5567723 mAP recorded during training 2018-11-14 15:14:41 +00:00
Glenn Jocher 4bae1d0f75 updates 2018-11-09 17:03:26 +01:00
Glenn Jocher b1a2735338 updates 2018-11-09 16:58:32 +01:00
Glenn Jocher 5177f3e7a0 updates 2018-11-09 16:48:55 +01:00
Glenn Jocher 664cbaab09 Adam optimizer 2018-11-09 16:44:12 +01:00
Glenn Jocher 46a4de77cb updates 2018-11-08 12:29:35 +01:00
Glenn Jocher 2463030d6c updates 2018-11-08 12:28:19 +01:00
Glenn Jocher c8e4a19879 updates 2018-11-08 12:27:14 +01:00
Glenn Jocher a6d69cefe0 updates 2018-11-08 12:26:23 +01:00
Glenn Jocher edfad8095d updates 2018-11-05 23:34:26 +01:00
Glenn Jocher 6e5da1ce27 updates 2018-11-05 23:32:36 +01:00
Glenn Jocher 19ccb41eaf updates 2018-11-05 23:28:10 +01:00
Glenn Jocher 0096bb4dd5 update multi-scale training 2018-11-05 23:20:21 +01:00
Glenn Jocher 77469a5268 update multi-scale training 2018-11-05 23:17:53 +01:00
Glenn Jocher 587097affb update LR scheduler 2018-11-05 09:20:18 +01:00
Glenn Jocher 2ccf68cf96 add multi_scale train option to argparser 2018-11-05 09:08:48 +01:00
Glenn Jocher dc7b58bb3c add multi_scale train option to argparser 2018-11-05 09:07:15 +01:00
Glenn Jocher 741626c55b initialize from darknet53 2018-10-30 15:20:52 +01:00
Glenn Jocher 26c52f9485 initialize from darknet53 2018-10-30 15:18:52 +01:00
Glenn Jocher ed0390d0b5 initialize from darknet53 2018-10-30 14:58:56 +01:00
Glenn Jocher 332fe002b3 rename /checkpoints to /weights 2018-10-30 14:58:26 +01:00
Glenn Jocher 0ae90d0fb7 rename /checkpoints to /weights 2018-10-27 00:42:34 +02:00
Glenn Jocher 05f28ab02b -batch_size from 12 to 16 2018-10-15 21:05:24 +02:00
Glenn Jocher f79e7ffa76 updates 2018-10-10 16:16:17 +02:00
Glenn Jocher d748bedb1d clean up train.py 2018-10-09 19:32:42 +02:00
Glenn Jocher e7cd5d01c4 cleanup train.py 2018-10-09 19:28:27 +02:00
Glenn Jocher 0cc8f2be01 clean up train.py 2018-10-09 19:22:33 +02:00
Glenn Jocher c01b8e6b7c updates 2018-10-05 16:38:59 +02:00
Glenn Jocher 0058431e2e create step lr schedule 2018-09-28 14:26:46 +02:00
Glenn Jocher 208fd77fe4 create step lr schedule 2018-09-25 01:29:35 +02:00
Glenn Jocher 6528238953 align loss to darknet 2018-09-24 21:26:12 +02:00
Glenn Jocher 396a71001e align loss to darknet 2018-09-24 21:25:17 +02:00
Glenn Jocher 750f528bfe align loss to darknet 2018-09-24 03:34:12 +02:00
Glenn Jocher 313a3f6b0c updates 2018-09-24 03:06:04 +02:00
Glenn Jocher a722601ef6 Adam to SGD with burn-in 2018-09-20 18:03:19 +02:00
Glenn Jocher 1cfde4aba8 nGT to nT 2018-09-19 04:32:16 +02:00
Glenn Jocher 29fbcb059f simplify train.py 2018-09-19 04:21:46 +02:00
Glenn Jocher b04ee34035 updates 2018-09-04 14:41:52 +02:00
Glenn Jocher 3a0c16fbc2 updates 2018-09-04 14:36:51 +02:00
Glenn Jocher aa77cbea11 updates 2018-09-03 00:44:55 +02:00
Glenn Jocher defae83d77 updates 2018-09-03 00:35:46 +02:00
Glenn Jocher caeba13b84 updates 2018-09-02 14:01:31 +02:00
Glenn Jocher 641e354948 updates 2018-09-02 11:38:39 +02:00
Glenn Jocher eeb546ed6f updates 2018-09-01 19:13:12 +02:00
Glenn Jocher 7d083f558a updates 2018-09-01 18:47:08 +02:00
Glenn Jocher 8fd8d8eb04 updates 2018-09-01 18:41:05 +02:00
Glenn Jocher 54d1da904c updates 2018-08-27 00:00:25 +02:00
Glenn Jocher ebe27544eb updates 2018-08-26 20:30:47 +02:00
Glenn Jocher 56badeef8a updates 2018-08-26 19:40:30 +02:00
Glenn Jocher af7144ba79 updates 2018-08-26 19:38:37 +02:00
Glenn Jocher 8a1d1b76c0 updates 2018-08-26 19:33:37 +02:00
Glenn Jocher ad0860dbe2 updates 2018-08-26 17:09:10 +02:00
Glenn Jocher 3fb6cc8161 updates 2018-08-26 15:40:07 +02:00
Glenn Jocher 184db1fb10 updates 2018-08-26 11:48:58 +02:00
Glenn Jocher 119d39599e updates 2018-08-26 11:48:19 +02:00
Glenn Jocher e81ef205fe updates 2018-08-26 11:44:41 +02:00
Glenn Jocher 7f2df90277 updates 2018-08-26 11:42:34 +02:00
Glenn Jocher c3731591af Initial commit 2018-08-26 10:51:39 +02:00