Commit Graph

576 Commits

Author SHA1 Message Date
Glenn Jocher e2e3e35e07 updates 2019-05-25 14:43:07 +02:00
Glenn Jocher 6f05c5347e updates 2019-05-25 11:57:19 +02:00
Glenn Jocher 8f9ab337b0 updates 2019-05-23 13:19:49 +02:00
Glenn Jocher 001193b9c7 updates 2019-05-23 13:15:44 +02:00
Glenn Jocher e67cee4a0c updates 2019-05-23 12:27:46 +02:00
Glenn Jocher 19c7697434 updates 2019-05-21 17:37:34 +02:00
Glenn Jocher 520e58aa05 updates 2019-05-21 16:11:08 +02:00
Glenn Jocher d09db54cb0 updates 2019-05-21 16:03:29 +02:00
Glenn Jocher d2589fc5f7 updates 2019-05-21 15:00:16 +02:00
Glenn Jocher 0effcd02bf updates 2019-05-21 14:01:06 +02:00
Glenn Jocher 401a615d34 updates 2019-05-21 12:59:04 +02:00
Glenn Jocher b2abd46437 updates 2019-05-21 12:58:18 +02:00
Glenn Jocher c2950dbfb6 updates 2019-05-20 20:06:36 +02:00
Glenn Jocher 5aeef8fba7 updates 2019-05-20 17:50:12 +02:00
Glenn Jocher 13180b51ac updates 2019-05-20 16:16:43 +02:00
Glenn Jocher 8eced539e0 updates 2019-05-20 14:52:14 +02:00
Glenn Jocher a9daf244b9 updates 2019-05-19 18:03:10 +02:00
Glenn Jocher 68490520db updates 2019-05-19 17:28:14 +02:00
Glenn Jocher b034382c8b updates 2019-05-18 23:24:26 +02:00
Glenn Jocher 4b15644b46 updates 2019-05-14 12:59:12 +02:00
Glenn Jocher 48d4c5938d updates 2019-05-13 21:18:15 +02:00
Glenn Jocher bc19e89247 add *.jpeg support 2019-05-11 14:38:48 +02:00
Glenn Jocher ae03cf3eea add *.jpeg support 2019-05-10 15:16:02 +02:00
Glenn Jocher 31592c276f add *.jpeg support 2019-05-10 14:15:09 +02:00
ypw 9a13bb53c8 Update utils.py (#268) 2019-05-10 12:27:31 +02:00
Glenn Jocher 1a757524bf add *.jpeg support 2019-05-09 11:23:36 +02:00
Glenn Jocher 5580694970 updates 2019-05-08 17:29:23 +02:00
Glenn Jocher 40a5680671 updates 2019-05-08 13:31:49 +02:00
Glenn Jocher a8f0a3fede updates 2019-05-08 13:06:24 +02:00
Glenn Jocher fc6ea5b1fd updates 2019-05-06 16:37:41 +02:00
Glenn Jocher 2ebbe2f339 updates 2019-05-05 22:10:02 +02:00
Glenn Jocher f05042b660 updates 2019-05-05 21:49:04 +02:00
Glenn Jocher 439c5a9839 updates 2019-05-05 21:02:18 +02:00
Glenn Jocher 74d5496b5b updates 2019-05-05 21:00:50 +02:00
Glenn Jocher 6f1b011bd9 updates 2019-05-05 14:29:30 +02:00
Glenn Jocher 4156e83319 updates 2019-05-05 14:13:05 +02:00
Glenn Jocher a8f443518f updates 2019-05-05 13:44:12 +02:00
Glenn Jocher 7d857cda95 updates 2019-05-05 13:21:37 +02:00
Glenn Jocher 6316171f33 updates 2019-05-03 18:14:16 +02:00
Glenn Jocher 3c55d63a9d updates 2019-05-03 00:26:26 +02:00
Glenn Jocher b901441e76 updates 2019-05-02 23:56:58 +02:00
Glenn Jocher ae41d5855a updates 2019-05-02 15:47:27 +02:00
Glenn Jocher 587522ec56 updates 2019-04-29 17:58:18 +02:00
Glenn Jocher 94f954eba0 updates 2019-04-29 17:57:51 +02:00
Glenn Jocher 7652365b28 updates 2019-04-27 21:38:20 +02:00
Glenn Jocher ccfd44c2f8 updates 2019-04-27 18:36:19 +02:00
Glenn Jocher 1e3fb6566c updates 2019-04-27 18:15:27 +02:00
Glenn Jocher d25190e15b updates 2019-04-27 17:51:59 +02:00
Glenn Jocher 8f1becd55c updates 2019-04-27 17:49:22 +02:00
Glenn Jocher 55077b2770 updates 2019-04-27 17:44:26 +02:00
Glenn Jocher 7691e2e0f8 updates 2019-04-26 13:28:00 +02:00
Glenn Jocher c5ec86082d updates 2019-04-26 12:24:18 +02:00
Glenn Jocher cf54fa7468 updates 2019-04-25 22:47:31 +02:00
Glenn Jocher 3bb38215dd updates 2019-04-24 21:48:32 +02:00
Glenn Jocher 365d38bc0c updates 2019-04-24 21:41:18 +02:00
Glenn Jocher 3e71b8d48b updates 2019-04-24 21:37:32 +02:00
Glenn Jocher fa0acebe2a updates 2019-04-24 21:28:11 +02:00
Glenn Jocher 83793ffb2b updates 2019-04-24 21:23:54 +02:00
Glenn Jocher 387bdb010d updates 2019-04-24 19:56:04 +02:00
Glenn Jocher fbf0014cd6 updates 2019-04-24 17:13:36 +02:00
Glenn Jocher aa2df1eda7 updates 2019-04-24 17:02:52 +02:00
Glenn Jocher 20bb5f0a6a updates 2019-04-24 16:39:56 +02:00
Glenn Jocher 55aaf9a21e updates 2019-04-24 14:19:43 +02:00
Glenn Jocher 87a450c933 updates 2019-04-23 18:54:27 +02:00
Glenn Jocher e2b554ca12 updates 2019-04-23 18:53:36 +02:00
Glenn Jocher 8d653ede3a updates 2019-04-23 18:36:43 +02:00
Glenn Jocher 5f69861958 updates 2019-04-22 16:52:14 +02:00
Glenn Jocher 23cd4ecfa7 updates 2019-04-22 16:17:01 +02:00
Glenn Jocher e5d11c68ac updates 2019-04-22 14:59:39 +02:00
Glenn Jocher cf2caaad41 updates 2019-04-22 14:31:23 +02:00
Glenn Jocher 0bac735cc6 updates 2019-04-22 12:51:20 +02:00
Glenn Jocher cfe354064c updates 2019-04-21 21:07:01 +02:00
Glenn Jocher 2bfea0c980 updates 2019-04-21 20:35:11 +02:00
Glenn Jocher f9d25f6d24 hyperparameter updates 2019-04-19 20:41:18 +02:00
Glenn Jocher fad618b821 updates 2019-04-18 23:05:19 +02:00
Glenn Jocher 0d770e14df updates 2019-04-18 22:55:50 +02:00
Glenn Jocher cc50757d95 updates 2019-04-18 22:31:05 +02:00
Glenn Jocher 9b6347ac6c updates 2019-04-18 21:56:50 +02:00
Glenn Jocher 31dc2e2be5 updates 2019-04-18 17:06:09 +02:00
Glenn Jocher b7c5eb1503 updates 2019-04-18 16:07:47 +02:00
Glenn Jocher c5e58b6484 updates 2019-04-18 14:48:14 +02:00
Glenn Jocher 286257c5ac updates 2019-04-18 14:47:05 +02:00
Glenn Jocher 2089e4f4c8 updates 2019-04-18 14:33:32 +02:00
Glenn Jocher 9a440cfa15 updates 2019-04-18 02:13:04 +02:00
Glenn Jocher 0b8a28e3dd updates 2019-04-17 15:52:51 +02:00
Glenn Jocher 9c5524ba82 updates 2019-04-17 14:57:39 +02:00
Glenn Jocher 6204d83f3a updates 2019-04-16 23:57:43 +02:00
Glenn Jocher a70c9f87a9 updates 2019-04-16 13:17:48 +02:00
Glenn Jocher 100f443722 updates 2019-04-16 13:03:24 +02:00
Glenn Jocher a8fb235647 updates 2019-04-16 12:55:23 +02:00
Glenn Jocher b5ec9cb128 updates 2019-04-16 12:49:34 +02:00
Glenn Jocher 00cc02ba91 updates 2019-04-15 13:58:48 +02:00
Glenn Jocher 76bf9fceed updates 2019-04-15 13:57:07 +02:00
Glenn Jocher 1191dee71b updates 2019-04-15 13:55:52 +02:00
Glenn Jocher aeca7f72c4 updates 2019-04-13 20:38:00 +02:00
Glenn Jocher 95f3d8e043 updates 2019-04-13 20:11:08 +02:00
Glenn Jocher 95696d24c0 updates 2019-04-12 17:19:00 +02:00
Glenn Jocher 50df252c4b updates 2019-04-12 14:58:19 +02:00
IlyaOvodov 5ea92e7ee2 FIX: trainig fails if targets list is empty (#198)
* FIX: trainig fails if targets list is empty

* Update utils.py
2019-04-12 14:55:26 +02:00
Glenn Jocher bce3dd03e8 updates 2019-04-12 14:00:16 +02:00
Glenn Jocher c9a55a269b updates 2019-04-11 15:29:31 +02:00
Glenn Jocher e6e6fb6f57 updates 2019-04-11 12:47:35 +02:00
Glenn Jocher cbd5347cc3 updates 2019-04-11 12:41:07 +02:00
Glenn Jocher 835f975228 updates 2019-04-11 12:21:33 +02:00
Glenn Jocher 9c7dc10b7f updates 2019-04-10 16:51:58 +02:00
Glenn Jocher f0b4f9f4fb updates 2019-04-10 16:39:15 +02:00
Glenn Jocher 7709e8aa72 updates 2019-04-09 16:28:14 +02:00
Glenn Jocher 2ca4c9aaec updates 2019-04-09 13:39:17 +02:00
Glenn Jocher d8cbf9b7a7 updates 2019-04-09 13:21:39 +02:00
Glenn Jocher 3e85a4191a updates 2019-04-09 13:19:17 +02:00
Glenn Jocher 2e74f4be41 updates 2019-04-09 12:32:26 +02:00
Glenn Jocher 4b7eb0aec9 updates 2019-04-09 12:24:32 +02:00
Glenn Jocher 26b115c306 updates 2019-04-09 12:24:01 +02:00
Glenn Jocher 05881d0730 updates 2019-04-09 11:38:16 +02:00
Glenn Jocher 3825e99ee3 updates 2019-04-08 15:41:14 +02:00
Glenn Jocher c948629fd3 updates 2019-04-06 16:16:40 +02:00
Glenn Jocher d171596183 updates 2019-04-06 16:14:16 +02:00
Glenn Jocher 112f061f4e updates 2019-04-06 16:13:11 +02:00
Glenn Jocher cb352be02c updates 2019-04-05 15:34:42 +02:00
Glenn Jocher c36f1e990b updates 2019-04-02 22:54:32 +02:00
Glenn Jocher be10b75eb4 updates 2019-04-02 20:21:00 +02:00
Glenn Jocher 1457f66419 updates 2019-04-02 18:04:04 +02:00
Glenn Jocher 3f82380e12 updates 2019-04-02 15:09:13 +02:00
Glenn Jocher 01569d15e3 updates 2019-04-02 13:43:18 +02:00
Glenn Jocher bd32517528 updates 2019-04-01 20:27:11 +02:00
Glenn Jocher a76e8e3ee8 updates 2019-04-01 18:43:21 +02:00
Glenn Jocher 4f98fbde78 updates 2019-04-01 18:42:54 +02:00
Glenn Jocher 09b02d2029 updates 2019-03-31 19:57:44 +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
Fatih Baltacı cb67d64a5f Update datasets.py (#169) 2019-03-27 15:44:19 +01:00
Glenn Jocher 47eab968ab updates 2019-03-26 18:02:57 +01:00
Glenn Jocher 5fd702aead updates 2019-03-25 19:39:14 +01:00
Glenn Jocher bd440fa0c3 updates 2019-03-25 18:35:39 +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
perry0418 4884508110
Update utils.py
solve the multi-gpu training problem
2019-03-25 14:59:02 +08:00
Glenn Jocher 4114d5e9c9 updates 2019-03-22 17:53:06 +02:00
Glenn Jocher b31f8fb017 updates 2019-03-22 15:08:03 +02:00
Glenn Jocher 75d8cbdd5f updates 2019-03-22 14:56:43 +02:00
Glenn Jocher 3532ee038f
Update datasets.py 2019-03-22 14:52:58 +02:00
WannaSeaU cd188dbde6
Empty label file may cause index error 2019-03-22 18:59:09 +08:00
Glenn Jocher 20beee0c5b updates 2019-03-21 22:51:22 +02:00
Glenn Jocher 6aef4e6a78 updates 2019-03-21 22:41:12 +02:00
Glenn Jocher 8ebb4da5cc updates 2019-03-21 17:28:26 +02:00
Glenn Jocher d047062074 updates 2019-03-21 17:26:31 +02:00
Glenn Jocher a024286ec1 updates 2019-03-21 15:05:20 +02:00
Glenn Jocher 70fe2204b4 multi_thread dataloader 2019-03-21 14:48:40 +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 2cd6805063 updates 2019-03-21 02:28:30 +02:00
Glenn Jocher ca67e2353b updates 2019-03-21 01:57:16 +02:00
Glenn Jocher 9885903baf updates 2019-03-20 20:31:09 +02:00
Glenn Jocher e7075f2b23 updates 2019-03-20 19:30:10 +02:00
Glenn Jocher f8994e89ea updates 2019-03-20 19:20:54 +02:00
Glenn Jocher 613ce1be1a updates 2019-03-19 15:50:15 +02:00
Glenn Jocher 00e181a55a updates 2019-03-19 12:38:35 +02:00
Glenn Jocher feb5fcb16f multi_gpu multi_scale 2019-03-19 11:38:01 +02:00
Glenn Jocher 056eed2833 multi_gpu multi_scale 2019-03-19 10:38:32 +02:00
Glenn Jocher 32f1def48f multi_gpu 2019-03-18 00:59:24 +02:00
Glenn Jocher 7dba5d0171 multi_gpu 2019-03-18 00:48:56 +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 2df8d7e9f6
nms speedup 2019-03-15 20:40:37 +02:00
Glenn Jocher a2ad00d6fc updates 2019-03-05 17:10:34 +01:00
Glenn Jocher 3e6800fbc9 updates 2019-03-05 16:13:40 +01:00
Glenn Jocher 54b62f5302 updates 2019-03-04 17:36:41 +01:00
Glenn Jocher 3bea4da604 updates 2019-03-04 17:34:53 +01:00
Glenn Jocher 3f21d5bb2e updates 2019-03-04 16:13:31 +01:00
Glenn Jocher dc9f2ef6ba updates 2019-03-04 16:11:37 +01:00
Glenn Jocher 5ca987eaed updates 2019-03-04 16:01:23 +01:00
Glenn Jocher 175e231c55 updates 2019-03-04 15:59:11 +01:00
Glenn Jocher 8069fc452a updates 2019-03-04 15:52:18 +01:00
Glenn Jocher 65ed0d1122 updates 2019-03-04 15:46:25 +01:00
Glenn Jocher 66211d1147 updates 2019-03-04 15:41:22 +01:00
Glenn Jocher 545f756090 updates 2019-02-28 15:40:30 +01:00
Glenn Jocher 55c6efbb39 updates 2019-02-27 14:46:28 +01:00
Glenn Jocher 303eef1d3d updates 2019-02-27 14:45:39 +01:00
Glenn Jocher e094bb14ba updates 2019-02-27 14:19:57 +01:00
Glenn Jocher 324dc6af6e updates 2019-02-27 13:21:39 +01:00
Glenn Jocher 41d55d452b updates 2019-02-27 12:52:02 +01:00
Glenn Jocher 036e3b3253 updates 2019-02-27 12:51:24 +01:00
Glenn Jocher 358f34afa8 updates 2019-02-27 12:32:25 +01:00
Glenn Jocher 9a27339e04 updates 2019-02-27 00:04:41 +01:00
Glenn Jocher a6fdc7413b updates 2019-02-26 15:12:21 +01:00
Glenn Jocher 57417e7080 updates 2019-02-26 15:00:27 +01:00
Glenn Jocher cb63ce30ec updates 2019-02-26 14:57:28 +01:00
Glenn Jocher 6814e925b5 updates 2019-02-26 03:12:32 +01:00
Glenn Jocher 41ba6dfd6b updates 2019-02-26 03:12:04 +01:00
Glenn Jocher 90a20f93e5 updates 2019-02-26 02:53:11 +01:00
Glenn Jocher f541861533 updates 2019-02-25 13:50:01 +01:00
Glenn Jocher d2cd49f059 updates 2019-02-25 13:47:51 +01:00
Glenn Jocher 3f68a6776a updates 2019-02-21 20:45:53 +01:00
Glenn Jocher a92b6d4d32 updates 2019-02-20 23:21:42 +01:00
Glenn Jocher ead4af98b0 updates 2019-02-20 15:11:55 +01:00
Glenn Jocher 315fe6ec14 xy and wh losses respectively merged 2019-02-20 12:50:40 +01:00
Glenn Jocher 0772ebf7c9 xy and wh losses respectively merged 2019-02-19 22:19:59 +01:00
Glenn Jocher 3eb49be263 xy and wh losses respectively merged 2019-02-19 20:03:42 +01:00
Glenn Jocher 15bba5a345 xy and wh losses respectively merged 2019-02-19 19:55:33 +01:00
Glenn Jocher 9df279cded updates 2019-02-19 19:36:09 +01:00
Glenn Jocher 9c96e7b6cd updates 2019-02-19 19:00:44 +01:00
Glenn Jocher ce4ee36ca0 updates 2019-02-18 19:58:01 +01:00
Glenn Jocher d81838e286 updates 2019-02-18 19:53:38 +01:00
Glenn Jocher bbb750876e updates 2019-02-18 19:52:38 +01:00
Glenn Jocher 0f06fbd681 updates 2019-02-18 19:49:58 +01:00
Glenn Jocher 2ef92f5651 updates 2019-02-18 19:44:15 +01:00
Glenn Jocher f788a57009 updates 2019-02-18 19:31:00 +01:00
Glenn Jocher 2ba45e4878 updates 2019-02-18 19:27:34 +01:00
Glenn Jocher 5f2d3aa9c3 updates 2019-02-18 19:24:53 +01:00
Glenn Jocher 77ce2cd43f updates 2019-02-18 19:21:21 +01:00
Glenn Jocher adea337545 updates 2019-02-18 19:17:48 +01:00
Glenn Jocher a80b2d1611 updates 2019-02-18 19:13:40 +01:00
Glenn Jocher e4d62de5bc updates 2019-02-18 18:32:31 +01:00
Glenn Jocher 6e2cf074a1 updates 2019-02-18 17:48:35 +01:00
Glenn Jocher c535a8699a updates 2019-02-18 15:51:35 +01:00
Glenn Jocher 6deda82384 updates 2019-02-18 14:03:39 +01:00
Glenn Jocher c828f5459f select GPU0 if multiple available 2019-02-16 14:33:52 +01:00
Glenn Jocher cc5e9a5a85 updates 2019-02-12 16:58:07 +01:00
Glenn Jocher e23b1a3d73 webcam updates 2019-02-11 18:15:51 +01:00
Glenn Jocher 585f2e2cc1 updates 2019-02-11 17:25:32 +01:00
Glenn Jocher 786e10a197 class labeling corrections 2019-02-11 13:45:04 +01:00
Glenn Jocher 1ca352b328 class labeling corrections 2019-02-11 12:44:12 +01:00
Glenn Jocher ebd682b25c updates 2019-02-11 12:40:14 +01:00
Glenn Jocher 429fd6121c updates 2019-02-11 12:26:46 +01:00
Glenn Jocher 3cd76b2185 updates 2019-02-10 23:27:31 +01:00
Glenn Jocher 22dc8c0ea6 updates 2019-02-10 22:01:53 +01:00
Glenn Jocher c60bad8b10 updates 2019-02-10 21:23:58 +01:00
Glenn Jocher 9d12a162f8 updates 2019-02-10 21:01:49 +01:00
Glenn Jocher e057f52780 updates 2019-02-10 20:32:04 +01:00
Glenn Jocher a701374014 updates 2019-02-09 19:24:51 +01:00
Glenn Jocher 334660d58f updates 2019-02-08 22:55:01 +01:00
Glenn Jocher c2436d8197 updates 2019-02-08 22:43:05 +01:00
Josh Veitch-Michaelis 212597fddd
Fix absolute path in class name loader 2019-01-06 20:54:04 +00:00
Glenn Jocher 8dfa653942 updates 2019-01-06 15:58:41 +02:00
Glenn Jocher 6e1ff541c9 updates 2019-01-06 14:23:04 +02:00
Glenn Jocher d673b6c5f4 updates 2019-01-03 23:44:51 +01:00
Glenn Jocher b181c61f4b updates 2019-01-02 16:32:38 +01:00
Glenn Jocher 7283f26f6f updates 2019-01-01 17:52:45 +01:00
Glenn Jocher 36a06a1e90 updates 2018-12-31 12:31:38 +01:00
Glenn Jocher cc018c73ad ONNX compatibility updates 2018-12-28 21:12:31 +01:00
Glenn Jocher 16bc3b72c3 updates 2018-12-28 20:11:10 +01:00
Glenn Jocher 4c5f4864fb ONNX compatibility updates 2018-12-26 15:57:18 +01:00
Glenn Jocher 8b34fbef33 ONNX compatibility updates 2018-12-26 15:46:39 +01:00
Glenn Jocher fb4383f364 ONNX export compatability updates 2018-12-23 13:46:47 +01:00
Glenn Jocher 62c186da25 updates 2018-12-19 23:48:52 +01:00
Glenn Jocher b079c1b10c updates 2018-12-15 21:01:14 +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 e28ac3de29 updates 2018-12-11 20:46:46 +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 be8603b2dd updates 2018-12-04 19:17:03 +01:00
Glenn Jocher 10cca39934 updates 2018-12-03 21:08:45 +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 5843c41dfc add multi_scale support 2018-12-03 14:05:50 +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 bd649f241f updates 2018-11-29 12:12:48 +01:00
Glenn Jocher 5a0575af3a updates 2018-11-27 18:43:46 +01:00
Glenn Jocher ab9ee6aa9a updates 2018-11-23 19:45:39 +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 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 b9d87be318 updates 2018-11-22 14:54:52 +01:00
Nir Ben-Zvi d41f85702d Fixed NMS bug causing big CPU usage. Note that using 'cross_class_nms' still takes a huge amount of time and should be fixed somehow. 2018-11-22 15:36:14 +02:00
Glenn Jocher f18f288990 updates 2018-11-22 14:29:50 +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 8ac8d0a382 updates 2018-11-21 19:22:35 +01:00
Glenn Jocher dae9b8f4b5 updates 2018-11-21 19:22:01 +01:00
Glenn Jocher 1415a798fe updates 2018-11-17 12:54:44 +01:00
Glenn Jocher dd7c3d2455 updates 2018-11-17 12:51:22 +01:00
Glenn Jocher 45c5567723 mAP recorded during training 2018-11-14 15:14:41 +00:00
Glenn Jocher 9dbc3ec1c4 updates 2018-11-13 11:20:01 +00:00
Glenn Jocher 34bc12d2ad updates 2018-11-11 18:58:41 +01:00
Glenn Jocher 538e5741c6 updates 2018-11-08 12:42:39 +01:00
Glenn Jocher a6d69cefe0 updates 2018-11-08 12:26:23 +01:00
Glenn Jocher 8d7660b438 updates 2018-11-07 15:17:00 +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 3afb29ad48 update multi-scale training 2018-11-05 23:20:45 +01:00
Glenn Jocher 77469a5268 update multi-scale training 2018-11-05 23:17:53 +01:00
Glenn Jocher 0c89122aab updates 2018-11-04 18:19:07 +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 24a41972cb BCE to CE lconf + batch size 16 2018-10-11 17:43:34 +02:00
Glenn Jocher d336e0053d per-class mAP report 2018-10-10 17:07:21 +02:00
Glenn Jocher f79e7ffa76 updates 2018-10-10 16:16:17 +02:00
Glenn Jocher 0cc8f2be01 clean up train.py 2018-10-09 19:22:33 +02:00
Glenn Jocher 07ac4fef8d create step lr schedule 2018-10-05 17:01:07 +02:00
Glenn Jocher c01b8e6b7c updates 2018-10-05 16:38:59 +02:00
Glenn Jocher 8a87521044 P and R conf thresh to 0.5 2018-10-05 16:01:27 +02:00
Glenn Jocher bce94f6ade corrected numpy printoptions 2018-10-03 13:55:56 +02:00
Glenn Jocher 5d402ad31a reapply yolo width and height 2018-09-23 22:41:36 +02:00
Glenn Jocher cf9b4cfa52 update loss components 2018-09-23 22:25:23 +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 a8f8ec134b updates 2018-09-10 17:00:39 +02:00
Glenn Jocher 300e2b5dfc updates 2018-09-10 16:41:02 +02:00
Glenn Jocher ff04315f96 updates 2018-09-10 16:35:00 +02:00
Glenn Jocher 34144aabe3 updates 2018-09-10 16:31:56 +02:00