Commit Graph

614 Commits

Author SHA1 Message Date
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
Glenn Jocher ba1b3d8fe5 updates 2018-09-10 16:19:00 +02:00
Glenn Jocher c1492ae4fb updates 2018-09-10 15:50:37 +02:00
Glenn Jocher e7dab5a42f mAP corrected to per-class 2018-09-10 15:12:13 +02:00
Glenn Jocher 6116acb8c2 np.unique sorting correction 2018-09-09 16:14:24 +02:00
Glenn Jocher a284fc921d updates 2018-09-08 14:46:22 +02:00
Glenn Jocher 0bfc4bcee3 updates 2018-09-04 15:08:32 +02:00
Glenn Jocher 966d85ba01 updates 2018-09-04 15:08:06 +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 58f2d9306b updates 2018-09-02 12:40:29 +02:00
Glenn Jocher 641e354948 updates 2018-09-02 11:38:39 +02:00
Glenn Jocher e99bda0c54 updates 2018-09-02 11:26:56 +02:00
Glenn Jocher 8ed89d8c88 updates 2018-09-02 11:15:39 +02:00
Glenn Jocher 7fa8dd9257 updates 2018-09-01 13:17:21 +02:00
Glenn Jocher c09703f4d4 updates 2018-09-01 13:11:57 +02:00
Glenn Jocher a660211733 updates 2018-08-26 23:59:13 +02:00
Glenn Jocher 6da62e433d updates 2018-08-26 20:24:37 +02:00
Glenn Jocher c0d1aad97e updates 2018-08-26 19:38:14 +02:00
Glenn Jocher 1bc4f89bca updates 2018-08-26 19:34:04 +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 b965b6e9b7 updates 2018-08-26 11:52:27 +02:00
Glenn Jocher 65228ba8a2 updates 2018-08-26 11:47:38 +02:00
Glenn Jocher a27276f055 updates 2018-08-26 11:30:46 +02:00
Glenn Jocher be17c9aaf6 updates 2018-08-26 10:59:39 +02:00
Glenn Jocher c3731591af Initial commit 2018-08-26 10:51:39 +02:00