Commit Graph

344 Commits

Author SHA1 Message Date
Glenn Jocher 926447e8c4 updates 2019-08-17 17:10:57 +02:00
Glenn Jocher b72fb74ad0 updates 2019-08-17 14:15:27 +02:00
Glenn Jocher a1200ef130 updates 2019-08-17 14:08:10 +02:00
Glenn Jocher 321bd95764 updates 2019-08-17 02:14:28 +02:00
Glenn Jocher 5a9fb2411d updates 2019-08-15 18:15:27 +02:00
Glenn Jocher 7fb64dbf67 updates 2019-08-12 13:49:38 +02:00
Glenn Jocher 4ac6e88ea9 memory-saving routs update 2019-08-12 13:37:11 +02:00
Glenn Jocher daaa8194a9 updates 2019-08-12 12:25:26 +02:00
Glenn Jocher 89d0aa7164 updates 2019-08-12 00:43:04 +02:00
Glenn Jocher 4e9a8661b2 updates 2019-08-11 15:22:53 +02:00
Glenn Jocher 636c1cff7a updates 2019-08-11 15:17:40 +02:00
Glenn Jocher e8a15ac1d7 updates 2019-08-10 22:11:55 +02:00
晨太狼 d41b444a15 Fix fuse (#440)
Fix fuse in models.py
2019-08-09 12:44:47 +02:00
Glenn Jocher c49fe688b7 updates 2019-08-08 19:49:15 +02:00
Glenn Jocher 1ddde51b80 updates 2019-08-08 18:48:37 +02:00
Glenn Jocher 141032045b updates 2019-08-06 16:57:33 +02:00
Glenn Jocher 658ef8a272 updates 2019-08-03 14:49:38 +02:00
Glenn Jocher 2d8311a83f updates 2019-08-03 14:38:06 +02:00
Glenn Jocher cd1f1eeecc updates 2019-08-03 14:22:25 +02:00
Glenn Jocher 90daf8f19c updates 2019-08-03 14:14:10 +02:00
Yonghye Kwon 333cf92bb2 Replace EmptyLayer with nn.Sequential (#420)
* Replace EmptyLayer with nn.Sequential

* Update models.py

* Update models.py
2019-08-03 13:24:14 +02:00
Yonghye Kwon 2a284aacd0 Cleanup- edit for Readability (#417)
Cleanup- edit for Readability
2019-08-02 14:49:08 +02:00
Glenn Jocher 79111cb18f updates 2019-08-01 21:58:43 +02:00
Glenn Jocher d07b9988e3 updates 2019-08-01 00:33:17 +02:00
Glenn Jocher 272b9c7c11 updates 2019-07-29 23:37:12 +02:00
Glenn Jocher 4af819449c updates 2019-07-29 12:06:29 +02:00
Glenn Jocher 3a7711856e Merge remote-tracking branch 'origin/master' 2019-07-29 00:45:37 +02:00
Glenn Jocher 9cf7f2215a updates 2019-07-29 00:42:03 +02:00
Glenn Jocher 88eea8f147 updates 2019-07-25 18:18:40 +02:00
Glenn Jocher 7b6cba86ef updates 2019-07-25 13:19:26 +02:00
Glenn Jocher a39ee4d252 updates 2019-07-20 17:05:09 +02:00
Glenn Jocher d6edefa8ab updates 2019-07-20 01:28:29 +02:00
Glenn Jocher 954deadff3 updates 2019-07-16 01:03:15 +02:00
Glenn Jocher bb1e551150 updates 2019-07-08 19:26:46 +02:00
Glenn Jocher 68b50f5cb6 updates 2019-07-08 12:43:15 +02:00
glenn-jocher ab141fcc1f updates 2019-07-03 15:37:04 +02:00
glenn-jocher 1d0a4a3ace updates 2019-07-03 14:42:11 +02:00
Glenn Jocher eeae43c414 updates 2019-06-28 00:38:52 +02:00
Glenn Jocher c1bb037cbe Merge remote-tracking branch 'origin/master' 2019-06-26 11:28:06 +02:00
Glenn Jocher cbfc5a00e5 updates 2019-06-26 11:27:36 +02:00
Yonghye Kwon 37fe87ccd9 cleanup- delete a variable "yolo_layer_count" (#347) 2019-06-26 11:10:52 +02:00
Glenn Jocher d7a28bd9f7 updates 2019-06-05 13:49:56 +02:00
Glenn Jocher f7a517d72c updates 2019-05-30 01:40:35 +02:00
Glenn Jocher c8a30663f0 updates 2019-05-14 18:43:14 +02:00
Glenn Jocher cc5660e7c0 updates 2019-05-14 18:37:13 +02:00
Glenn Jocher dd2d713484 updates 2019-05-03 20:51:30 +02:00
Glenn Jocher 09ee7b6f11 updates 2019-05-03 20:50:44 +02:00
Glenn Jocher 324f860235 updates 2019-04-25 20:50:37 +02:00
Glenn Jocher 85a4cf0042 updates 2019-04-23 16:48:47 +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 37799efa0b updates 2019-04-21 23:49:10 +02:00
Glenn Jocher cfe354064c updates 2019-04-21 21:07:01 +02:00
Glenn Jocher a6dc4347a3 updates 2019-04-21 20:36:04 +02:00
Glenn Jocher 14e4519620 updates 2019-04-21 20:30:11 +02:00
Glenn Jocher 4a4668224b Fuse Conv2d + BatchNorm2d 2019-04-20 22:46:23 +02:00
Glenn Jocher f9d25f6d24 hyperparameter updates 2019-04-19 20:41:18 +02:00
Glenn Jocher 97c488f8ef updates 2019-04-18 19:18:28 +02:00
Glenn Jocher 1d7ccb7580 updates 2019-04-18 19:17:08 +02:00
Glenn Jocher bce3dd03e8 updates 2019-04-12 14:00:16 +02:00
Glenn Jocher cbd5347cc3 updates 2019-04-11 12:41:07 +02:00
Glenn Jocher 7ee48a43b6 updates 2019-04-06 16:13:35 +02:00
Glenn Jocher cb352be02c updates 2019-04-05 15:34:42 +02:00
Glenn Jocher efc662351b updates 2019-04-03 17:24: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 70fe2204b4 multi_thread dataloader 2019-03-21 14:48:40 +02:00
Glenn Jocher 973715060d multi_gpu multi_scale 2019-03-19 15:48:52 +02:00
Glenn Jocher bc989a0147 multi_gpu multi_scale 2019-03-19 15:44:36 +02:00
Glenn Jocher 735b1a370b multi_gpu multi_scale 2019-03-19 15:43:10 +02:00
Glenn Jocher 2f1afd2d69 multi_gpu multi_scale 2019-03-19 15:38:53 +02:00
Glenn Jocher dcdd1ae6b7 multi_gpu multi_scale 2019-03-19 15:35: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
Daniel Suess 10182ca39d Get rid of hardcoded values of 85 2019-03-13 13:00:25 +11:00
Daniel Suess 003de1917e Fix shape-mismatch in ONNX export 2019-03-13 12:50:13 +11:00
Glenn Jocher 473eb8d0c9
Update models.py 2019-03-05 18:43:51 +01:00
Glenn Jocher 3bea4da604 updates 2019-03-04 17:34:53 +01:00
Glenn Jocher dc9f2ef6ba updates 2019-03-04 16:11:37 +01:00
Glenn Jocher 5fcdcefec3 updates 2019-03-04 16:06:13 +01:00
Glenn Jocher 175e231c55 updates 2019-03-04 15:59:11 +01:00
Glenn Jocher 0b3a17362c updates 2019-03-04 15:56:58 +01:00
Glenn Jocher 5e0cce771e updates 2019-03-04 15:52:51 +01:00
Glenn Jocher 8069fc452a updates 2019-03-04 15:52:18 +01:00
Glenn Jocher 46e3343494 updates 2019-02-21 16:16:35 +01:00
Glenn Jocher af853f604c updates 2019-02-21 15:57:18 +01:00
Glenn Jocher ec308d605e updates 2019-02-21 10:59:05 +01:00
Glenn Jocher 646a21a5cd updates 2019-02-21 10:57:55 +01:00
Glenn Jocher 58d4826a11 updates 2019-02-20 23:52:36 +01:00
Glenn Jocher ed37551c38 updates 2019-02-20 12:55:06 +01:00
Glenn Jocher 344bea20eb updates 2019-02-20 12:53:38 +01:00
Glenn Jocher f728bd21d2 updates 2019-02-20 12:52:39 +01:00
Glenn Jocher 315fe6ec14 xy and wh losses respectively merged 2019-02-20 12:50:40 +01:00
Glenn Jocher 44333ddd9f updates 2019-02-19 22:49:47 +01:00
Glenn Jocher 15bba5a345 xy and wh losses respectively merged 2019-02-19 19:55:33 +01:00
Glenn Jocher 75225e4d99 updates 2019-02-19 19:30:56 +01:00
Glenn Jocher a116dd36f7 updates 2019-02-19 19:18:03 +01:00
Glenn Jocher f07dd72a09 updates 2019-02-19 19:01:31 +01:00
Glenn Jocher 9c96e7b6cd updates 2019-02-19 19:00:44 +01:00
Glenn Jocher 0dd791b7ad updates 2019-02-19 16:11:18 +01:00
Glenn Jocher e4d62de5bc updates 2019-02-18 18:32:31 +01:00
Glenn Jocher fa0cbca69a updates 2019-02-18 17:47:30 +01:00
Glenn Jocher 8de043980a updates 2019-02-18 16:25:57 +01:00
Glenn Jocher c535a8699a updates 2019-02-18 15:51:35 +01:00
Glenn Jocher 03685866fd updates 2019-02-17 18:04:23 +01:00
Glenn Jocher e919635564 updates 2019-02-17 18:02:56 +01:00
Glenn Jocher 9086caf0bb updates 2019-02-16 14:47:16 +01:00
Glenn Jocher ee4abc8cdf optimize imports 2019-02-15 14:08:05 +02:00
Glenn Jocher 9706002b71 optimize imports 2019-02-12 18:05:58 +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 22dc8c0ea6 updates 2019-02-10 22:01:53 +01:00
Glenn Jocher d5b17c93ff updates 2019-02-09 22:39:04 +01:00
Glenn Jocher 5ec27663e6 updates 2019-02-09 22:38:51 +01:00
Glenn Jocher f908f845ae updates 2019-02-09 22:14:07 +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 88804cad3b
Update models.py 2019-01-10 10:32:39 +01:00
Glenn Jocher 646a573740 updates 2019-01-09 11:48:04 +01:00
Glenn Jocher 558b23bca7 updates 2019-01-06 23:57:59 +01:00
Glenn Jocher 6e1ff541c9 updates 2019-01-06 14:23:04 +02:00
Glenn Jocher 5a7313ca5a updates 2019-01-03 23:42:07 +01:00
Glenn Jocher cff2a81315 updates 2019-01-03 23:41:31 +01:00
Glenn Jocher b181c61f4b updates 2019-01-02 16:32:38 +01:00
Glenn Jocher 17a02ae3e4 updates 2018-12-31 12:33:34 +01:00
Glenn Jocher 36a06a1e90 updates 2018-12-31 12:31:38 +01:00
Glenn Jocher d1951c1868 ONNX compatibility updates 2018-12-28 20:15:26 +01:00
Glenn Jocher 29f2e80950 ONNX compatibility updates 2018-12-26 12:33:34 +01:00
Glenn Jocher 6940221948 ONNX compatibility updates 2018-12-26 12:32:34 +01:00
Glenn Jocher 647e1c6f52 ONNX compatibility updates 2018-12-25 13:24:21 +01:00
Glenn Jocher b6ff9cad79 updates 2018-12-24 14:33:05 +01:00
Glenn Jocher 5403581e38 updates 2018-12-24 13:11:21 +01:00
Glenn Jocher c50df0d1db ONNX export compatability updates 2018-12-23 12:51:02 +01:00
Glenn Jocher bd5b9693bf enable yolov3-tiny inference 2018-12-22 12:36:33 +01:00
Glenn Jocher b52a49cf12 updates 2018-12-15 21:06:39 +01:00
Glenn Jocher 5843c41dfc add multi_scale support 2018-12-03 14:05:50 +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 35e445c5da updates 2018-11-29 22:10:35 +01:00
Glenn Jocher d5331be0a0 updates 2018-11-29 11:43:19 +01:00
Glenn Jocher b07ee41867 updates 2018-11-27 18:14:48 +01:00
Glenn Jocher 82124805f8 updates 2018-11-23 18:13:35 +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 154fae4430 updates 2018-11-22 15:04:02 +01:00
Glenn Jocher b9d87be318 updates 2018-11-22 14:54:52 +01:00
Glenn Jocher 959c67b4ed updates 2018-11-22 14:48:57 +01:00
Glenn Jocher bec94be01a updates 2018-11-22 14:33:01 +01: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 4eed25dad0 updates 2018-11-19 16:54:00 +01:00
Glenn Jocher e93fbf2338 updates 2018-11-19 13:15:15 +01:00
Glenn Jocher 6ab407231d updates 2018-11-19 12:55:03 +01:00
Glenn Jocher aa895d2a07 updates 2018-11-17 14:37:36 +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 07f15b68d3 updates 2018-11-17 00:32:28 +01:00
Glenn Jocher 0bc111f0b4 updates 2018-11-16 22:35:44 +01:00
Glenn Jocher ed1067bfb5 updates 2018-11-16 22:34:44 +01:00
Glenn Jocher d2c5d7a5fd updates 2018-11-16 20:01:38 +01:00
Glenn Jocher e04bb75ff1 updates 2018-11-10 00:54:55 +01:00
Glenn Jocher 966bc16d1a updates 2018-11-10 00:53:53 +01:00
Glenn Jocher 98484bbe2f updates 2018-11-09 17:18:37 +01:00
Glenn Jocher dc7b58bb3c add multi_scale train option to argparser 2018-11-05 09:07:15 +01:00
Glenn Jocher 332fe002b3 rename /checkpoints to /weights 2018-10-30 14:58:26 +01: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 8a87521044 P and R conf thresh to 0.5 2018-10-05 16:01:27 +02:00
Glenn Jocher ff630b1960 align loss to darknet 2018-09-25 03:45:52 +02:00
Glenn Jocher c09dc09dba align loss to darknet 2018-09-25 01:30:51 +02:00
Glenn Jocher a75119b8f0 align loss to darknet 2018-09-24 20:32:05 +02:00
Glenn Jocher 750f528bfe align loss to darknet 2018-09-24 03:34:12 +02:00
Glenn Jocher 292af1f2f4 align loss to darknet 2018-09-24 03:10:42 +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 bd3f617129 updates 2018-09-22 21:50:01 +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 68de92f1a1 loss lambda corrections 2018-09-13 14:15:49 +02:00
Glenn Jocher 9514e74438 updates 2018-09-10 17:02:38 +02:00
Glenn Jocher e2a8f5bdce updates 2018-09-02 12:59:39 +02:00
Glenn Jocher 058bb7f38d updates 2018-09-02 12:55:05 +02:00
Glenn Jocher 7500471526 updates 2018-08-27 00:01:41 +02:00
Glenn Jocher 2769d79d05 updates 2018-08-26 23:52:06 +02:00
Glenn Jocher 8944db80b9 updates 2018-08-26 22:37:23 +02:00
Glenn Jocher d378e32803 updates 2018-08-26 21:59:55 +02:00
Glenn Jocher d602aed1d8 updates 2018-08-26 21:41:35 +02:00
Glenn Jocher 8a1d1b76c0 updates 2018-08-26 19:33:37 +02:00
Glenn Jocher c3731591af Initial commit 2018-08-26 10:51:39 +02:00