Commit Graph

197 Commits

Author SHA1 Message Date
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 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 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 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
Glenn Jocher 47eab968ab updates 2019-03-26 18:02:57 +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 6aef4e6a78 updates 2019-03-21 22:41:12 +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 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 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 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 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 cb63ce30ec updates 2019-02-26 14:57:28 +01:00
Glenn Jocher 90a20f93e5 updates 2019-02-26 02:53:11 +01:00
Glenn Jocher 3f68a6776a updates 2019-02-21 20:45:53 +01:00
Glenn Jocher ead4af98b0 updates 2019-02-20 15:11:55 +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 e23b1a3d73 webcam updates 2019-02-11 18:15:51 +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 9d12a162f8 updates 2019-02-10 21:01:49 +01:00
Josh Veitch-Michaelis 212597fddd
Fix absolute path in class name loader 2019-01-06 20:54:04 +00: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 cc018c73ad ONNX compatibility updates 2018-12-28 21:12:31 +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 c591936446 updates 2018-12-11 21:49:56 +01:00
Guillermo García 5a566454f5 Extract seed and cuda initialization utils 2018-12-05 11:55:27 +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