Commit Graph

723 Commits

Author SHA1 Message Date
Glenn Jocher c413aaefea updates 2019-07-26 23:52:13 +02:00
Glenn Jocher 8615b3a7c3 updates 2019-07-26 19:11:59 +02:00
Glenn Jocher d0c46a2ca4 updates 2019-07-26 15:41:02 +02:00
Glenn Jocher 7a81cc2499 updates 2019-07-26 12:00:43 +02:00
Glenn Jocher bd92457604 updates 2019-07-25 17:50:19 +02:00
Glenn Jocher 8df3601663 updates 2019-07-25 17:49:54 +02:00
Glenn Jocher df4f25e610 updates 2019-07-25 13:51:55 +02:00
Glenn Jocher 7b6cba86ef updates 2019-07-25 13:19:26 +02:00
Glenn Jocher 169d117870 updates 2019-07-24 20:32:24 +02:00
Glenn Jocher db08ac5403 updates 2019-07-24 20:16:35 +02:00
Glenn Jocher ce8a597dab updates 2019-07-24 19:31:38 +02:00
Glenn Jocher 4fb7fbf4bc updates 2019-07-24 19:02:24 +02:00
Glenn Jocher e28437720d updates 2019-07-24 18:28:11 +02:00
Glenn Jocher ce4ea0c332 updates 2019-07-24 18:22:59 +02:00
Glenn Jocher 3cfc84a183 updates 2019-07-24 18:02:26 +02:00
Glenn Jocher 1cde55f7c9 updates 2019-07-24 15:56:10 +02:00
Glenn Jocher a8596c6af4 updates 2019-07-24 13:32:14 +02:00
Glenn Jocher 9c7fce0075 updates 2019-07-24 00:22:07 +02:00
Glenn Jocher 8a2986e38f updates 2019-07-23 15:45:14 +02:00
Glenn Jocher b025b3123e updates 2019-07-23 15:08:28 +02:00
Glenn Jocher 91eaf2f8fe updates 2019-07-23 01:35:03 +02:00
Glenn Jocher d20131b2e8 updates 2019-07-22 21:28:59 +02:00
Glenn Jocher 7fa6ad6b47 updates 2019-07-22 21:24:24 +02:00
Glenn Jocher 8ae16adde4 updates 2019-07-22 16:14:41 +02:00
Glenn Jocher 1cfd370455 updates 2019-07-22 02:18:34 +02:00
Glenn Jocher 3b108afac3 updates 2019-07-21 22:49:11 +02:00
Glenn Jocher a812b1f8b1 updates 2019-07-21 21:33:45 +02:00
Glenn Jocher 14cbd8c0ca updates 2019-07-21 19:18:15 +02:00
Glenn Jocher c9ed072cc7 updates 2019-07-21 05:06:58 +02:00
Glenn Jocher 9f7e71acab updates 2019-07-21 03:48:57 +02:00
Glenn Jocher 7d43ee0d02 updates 2019-07-21 03:39:01 +02:00
Glenn Jocher e92a8afae1 updates 2019-07-21 01:05:45 +02:00
Glenn Jocher deb200f6bf updates 2019-07-20 15:10:31 +02:00
Glenn Jocher 39f63b7110 updates 2019-07-20 15:04:41 +02:00
Glenn Jocher 4816969933 updates 2019-07-20 14:54:37 +02:00
Glenn Jocher 44b340321f updates 2019-07-20 13:20:01 +02:00
Glenn Jocher d6edefa8ab updates 2019-07-20 01:28:29 +02:00
Glenn Jocher 407a4c481d updates 2019-07-17 14:16:21 +02:00
Glenn Jocher 33838b558d updates 2019-07-17 14:14:42 +02:00
Glenn Jocher 153762dec0 updates 2019-07-16 18:58:49 +02:00
Glenn Jocher 64b606a3cd updates 2019-07-16 18:49:54 +02:00
Glenn Jocher 51d7e460a3 updates 2019-07-16 18:18:08 +02:00
Glenn Jocher 81540b80b9 updates 2019-07-16 18:06:24 +02:00
Glenn Jocher b459587cb0 updates 2019-07-16 17:56:39 +02:00
Glenn Jocher 813024116b updates 2019-07-16 17:50:41 +02:00
Glenn Jocher 034d2949b9 updates 2019-07-16 17:43:01 +02:00
Glenn Jocher 09b3670579 updates 2019-07-16 17:35:20 +02:00
Glenn Jocher 8501aed49f updates 2019-07-15 17:54:31 +02:00
Glenn Jocher 96e25462e8 updates 2019-07-15 17:00:04 +02:00
Glenn Jocher 6509d8e588 updates 2019-07-14 22:28:48 +02:00
Glenn Jocher 9c776b8052 updates 2019-07-14 21:38:55 +02:00
Glenn Jocher 3fc676b28a updates 2019-07-14 11:29:07 +02:00
Glenn Jocher 831b6e39b6 updates 2019-07-12 17:02:04 +02:00
Glenn Jocher 03c6fe1ffe updates 2019-07-12 16:10:37 +02:00
Glenn Jocher 0aa9759a90 updates 2019-07-12 15:44:39 +02:00
Glenn Jocher bb38391342 updates 2019-07-12 14:28:46 +02:00
Glenn Jocher bd9789aa00 equal layer weights 2019-07-12 12:23:17 +02:00
Glenn Jocher 5886200401 updates 2019-07-12 01:19:32 +02:00
Glenn Jocher a2909c59f8 updates 2019-07-11 11:57:10 +02:00
Glenn Jocher b005a17eff updates 2019-07-11 11:56:46 +02:00
Glenn Jocher 3373006d0e updates 2019-07-10 22:11:48 +02:00
Glenn Jocher 4f6ef59d92 updates 2019-07-10 20:47:05 +02:00
Glenn Jocher a9e42a16f1 updates 2019-07-10 19:48:29 +02:00
Glenn Jocher bb1e551150 updates 2019-07-08 19:26:46 +02:00
Glenn Jocher 0bd763f528 updates 2019-07-08 18:32:31 +02:00
Glenn Jocher feeaf734f2 updates 2019-07-08 18:04:44 +02:00
Glenn Jocher da9ec7d12f updates 2019-07-08 18:00:19 +02:00
Glenn Jocher 60bc2c1fbd updates 2019-07-08 15:43:46 +02:00
Glenn Jocher 94669fb704 updates 2019-07-08 15:24:20 +02:00
Glenn Jocher 291c3ec9c7 updates 2019-07-08 15:02:20 +02:00
glenn-jocher 70f6379601 GIoU to default 2019-07-07 23:24:34 +02:00
glenn-jocher 32a52dfb02 GIoU to default 2019-07-05 12:33:37 +02:00
glenn-jocher 429bd3b8a9 GIoU to default 2019-07-05 11:41:43 +02:00
glenn-jocher b649a95c9a GIoU to default 2019-07-05 00:36:37 +02:00
glenn-jocher abf59f1565 updates 2019-07-04 22:10:46 +02:00
glenn-jocher d0eace6cec updates 2019-07-04 21:34:33 +02:00
glenn-jocher 109991198c updates 2019-07-03 16:18:08 +02:00
glenn-jocher 1e62ee2152 updates 2019-07-03 16:17:46 +02:00
glenn-jocher ab141fcc1f updates 2019-07-03 15:37:04 +02:00
glenn-jocher a8cf64af31 updates 2019-07-02 18:21:28 +02:00
Yonghye Kwon ccf757b3ea changed the criteria for the best weight file (#356)
* changed the criteria for the best weight file

changed the criteria for the best weight file from loss to mAP

I trained the model on my custom dataset. But I failed to get a good results when I load the weight file that has the lowest loss on test dataset. 

I thought that the loss used in YOLO is not proper criteria for detection performance. So I changed the criteria from loss to mAP.

what do you think of this?

* Update train.py
2019-07-02 12:24:18 +02:00
glenn-jocher 1fd871abd8 updates 2019-07-01 17:44:42 +02:00
glenn-jocher f43ee6ef94 updates 2019-07-01 17:17:29 +02:00
glenn-jocher cf51cf9c99 updates 2019-07-01 17:14:42 +02:00
glenn-jocher 05358accbb updates 2019-07-01 15:23:30 +02:00
glenn-jocher c4409aa2ed updates 2019-07-01 15:22:22 +02:00
glenn-jocher b0d62e5204 updates 2019-07-01 15:21:06 +02:00
glenn-jocher 5e2b802f68 updates 2019-07-01 14:48:44 +02:00
glenn-jocher 63036deeb7 updates 2019-07-01 00:41:13 +02:00
glenn-jocher 32f5ea955b updates 2019-06-30 17:47:10 +02:00
glenn-jocher db2674aa31 updates 2019-06-30 17:34:29 +02:00
glenn-jocher 388b66dcd0 updates 2019-06-30 15:24:34 +02:00
Glenn Jocher 45540c787f updates 2019-06-25 19:36:11 +02:00
Glenn Jocher 1d76751e1f updates 2019-06-25 12:37:24 +02:00
Glenn Jocher e4cc830690 updates 2019-06-25 12:21:27 +02:00
Glenn Jocher 9a56d97059 updates 2019-06-25 11:54:19 +02:00
Glenn Jocher 6b222df35d updates 2019-06-24 15:56:20 +02:00
Glenn Jocher d208f006a1 updates 2019-06-24 14:46:00 +02:00
Glenn Jocher 0005823d1f updates 2019-06-24 13:43:17 +02:00
Glenn Jocher 57b616b8b1 updates 2019-06-23 22:01:11 +02:00
Glenn Jocher ef3e1343e2 updates 2019-06-22 15:52:27 +02:00
Glenn Jocher f501a0fc9d updates 2019-06-22 15:50:04 +02:00
Glenn Jocher 4f7fee45ff updates 2019-06-21 21:27:50 +02:00
Glenn Jocher 5f6c2b3d12 updates 2019-06-21 13:19:23 +02:00
Glenn Jocher 1a9aa30efc updates 2019-06-21 11:57:26 +02:00
Glenn Jocher 3223c0171a updates 2019-06-21 10:58:12 +02:00
Glenn Jocher a7e21b4315 updates 2019-06-21 10:24:06 +02:00
Glenn Jocher 7d7d7a6332 updates 2019-06-21 10:17:29 +02:00
Glenn Jocher c3526e0eff updates 2019-06-18 17:35:47 +02:00
Glenn Jocher 1096596ad8 updates 2019-06-18 16:36:04 +02:00
Glenn Jocher 84c1fecd81 updates 2019-06-18 16:33:18 +02:00
Glenn Jocher 4fb2567aa5 updates 2019-06-18 16:32:37 +02:00
Glenn Jocher 6efb2c935f updates 2019-06-18 16:32:19 +02:00
Glenn Jocher f573250ae8 updates 2019-06-16 23:17:40 +02:00
Glenn Jocher b59532883a updates 2019-06-15 17:06:58 +02:00
Glenn Jocher 995dc3ca67 updates 2019-06-15 02:44:01 +02:00
Glenn Jocher 02291622fa updates 2019-06-15 02:10:15 +02:00
Glenn Jocher bb3682024e updates 2019-06-15 01:35:55 +02:00
Glenn Jocher 8f609246db updates 2019-06-13 18:13:30 +02:00
Glenn Jocher 1e8df4db23 updates 2019-06-12 18:21:42 +02:00
Glenn Jocher 64134706d1 updates 2019-06-12 15:14:13 +02:00
Glenn Jocher b33a0b6cf2 updates 2019-06-12 15:12:08 +02:00
Glenn Jocher c5cb3c8a9e updates 2019-06-12 14:35:10 +02:00
Glenn Jocher bca423ee43 updates 2019-06-12 14:15:28 +02:00
Glenn Jocher e42865a304 updates 2019-06-12 13:25:39 +02:00
Glenn Jocher ef1703f2b8 updates 2019-06-12 13:19:17 +02:00
Glenn Jocher b5630f145f updates 2019-06-12 13:10:31 +02:00
Glenn Jocher 8df215a8cc updates 2019-06-12 13:04:58 +02:00
Glenn Jocher 64933f7ce0 updates 2019-06-12 11:55:20 +02:00
Glenn Jocher 5edb0ec40d updates 2019-06-12 11:50:24 +02:00
Glenn Jocher 9c328b1b0e updates 2019-06-12 11:41:44 +02:00
Glenn Jocher fe9cba308d updates 2019-06-12 11:41:00 +02:00
Glenn Jocher dd7ca339f5 updates 2019-06-12 11:40:17 +02:00
Glenn Jocher b0b6554eee updates 2019-06-12 11:37:25 +02:00
Glenn Jocher d5e2daf79d updates 2019-06-12 11:36:32 +02:00
Glenn Jocher 051b251d41 updates 2019-06-12 11:35:53 +02:00
Glenn Jocher 01e11aee04 updates 2019-06-12 11:30:37 +02:00
Glenn Jocher 37a07a44a1 updates 2019-06-12 11:29:36 +02:00
Glenn Jocher a2d392b5c3 updates 2019-06-12 11:25:56 +02:00
Glenn Jocher 7807337c5f updates 2019-05-31 14:30:27 +02:00
Glenn Jocher a2c9cb9d2c updates 2019-05-31 01:33:17 +02:00
Glenn Jocher ea11af6132 updates 2019-05-30 20:21:25 +02:00
Glenn Jocher 70ba7f0805 updates 2019-05-30 19:08:43 +02:00
Glenn Jocher 504d3b3f71 updates 2019-05-30 19:02:55 +02:00
Glenn Jocher 001193b9c7 updates 2019-05-23 13:15:44 +02:00
Glenn Jocher 68b9df4dd4 updates 2019-05-23 12:36:13 +02:00
Glenn Jocher 3006c33c29 updates 2019-05-23 12:32:11 +02:00
Glenn Jocher e67cee4a0c updates 2019-05-23 12:27:46 +02:00
Glenn Jocher 463dc56f31 updates 2019-05-21 17:39:04 +02:00
Glenn Jocher 19c7697434 updates 2019-05-21 17:37:34 +02:00
Glenn Jocher d09db54cb0 updates 2019-05-21 16:03:29 +02:00
Glenn Jocher b2abd46437 updates 2019-05-21 12:58:18 +02:00
Glenn Jocher b034382c8b updates 2019-05-18 23:24:26 +02:00
Glenn Jocher aae93a9651 updates 2019-05-18 12:07:57 +02:00
Glenn Jocher e42278e981 updates 2019-05-13 22:01:10 +02:00
Glenn Jocher 584e0a3be8 add *.jpeg support 2019-05-11 14:55:10 +02:00
Glenn Jocher c45cdc4fa3 updates 2019-05-10 17:08:00 +02:00
Glenn Jocher 9ffb40b0be updates 2019-05-10 16:29:37 +02:00
Glenn Jocher ddd0474111 add *.jpeg support 2019-05-10 15:28:54 +02:00
Glenn Jocher 3f8b64974c add *.jpeg support 2019-05-10 15:24:03 +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
Glenn Jocher 40bc015b75 updates 2019-05-08 18:54:55 +02:00
Glenn Jocher 5580694970 updates 2019-05-08 17:29:23 +02:00
Glenn Jocher e9246d8e63 updates 2019-05-08 14:25:30 +02:00
Glenn Jocher fa11c951ad updates 2019-05-08 14:25:13 +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 6316171f33 updates 2019-05-03 18:14:16 +02:00
Glenn Jocher b1604be04c updates 2019-04-29 17:49:09 +02:00
Glenn Jocher acaab77b7a updates 2019-04-27 17:58:16 +02:00
Glenn Jocher a9108a296b updates 2019-04-27 17:57:07 +02:00
Glenn Jocher 469dede6d1 updates 2019-04-27 17:52:43 +02:00
Glenn Jocher d25190e15b updates 2019-04-27 17:51:59 +02:00
Glenn Jocher 55077b2770 updates 2019-04-27 17:44:26 +02:00
Glenn Jocher 5263608d12 updates 2019-04-26 14:49:40 +02:00
Glenn Jocher 75f08c1cd1 updates 2019-04-26 13:56:44 +02:00
Glenn Jocher c89982d134 updates 2019-04-24 22:10:24 +02:00
Glenn Jocher 365d38bc0c updates 2019-04-24 21:41:18 +02:00
Glenn Jocher 83793ffb2b updates 2019-04-24 21:23:54 +02:00
Glenn Jocher 3a375f7132 updates 2019-04-24 14:09:15 +02:00
Glenn Jocher bd2378fad1 updates 2019-04-24 13:30:24 +02:00
Glenn Jocher 1771ffb1cf updates 2019-04-24 12:58:14 +02:00
Glenn Jocher 334c7c94cf updates 2019-04-22 23:27:31 +02:00
Glenn Jocher 2bfea0c980 updates 2019-04-21 20:35:11 +02:00
Glenn Jocher 5962510b23 hyperparameter updates 2019-04-19 13:18:47 +02:00
Glenn Jocher 6525c76f9c hyperparameter updates 2019-04-19 12:37:38 +02:00
Glenn Jocher 2643af18b2 updates 2019-04-18 23:42:37 +02:00
Glenn Jocher 9b6347ac6c updates 2019-04-18 21:56:50 +02:00
Glenn Jocher cf5bbc97ee updates 2019-04-18 21:44:57 +02:00
Glenn Jocher 03dd0b82ea updates 2019-04-18 21:42:21 +02:00
Glenn Jocher 02d6b2f9c5 updates 2019-04-18 19:25:41 +02:00
Glenn Jocher 023baa5aa2 updates 2019-04-18 17:07:41 +02:00
Glenn Jocher d822276cdb updates 2019-04-18 16:48:09 +02:00
Glenn Jocher 05ec8b3b9d updates 2019-04-18 16:45:38 +02:00
Glenn Jocher c68093ee7d updates 2019-04-18 16:17:44 +02:00
Glenn Jocher c711719280 updates 2019-04-18 15:56:31 +02:00
Glenn Jocher a4f2ad1660 updates 2019-04-18 15:39:05 +02:00
Glenn Jocher 25bf9e3611 updates 2019-04-18 15:31:03 +02:00
Glenn Jocher 8fcfb6ac3a updates 2019-04-18 15:24:58 +02:00
Glenn Jocher ee410481a0 updates 2019-04-18 15:18:09 +02:00
Glenn Jocher 48f6529fd1 updates 2019-04-18 15:17:31 +02:00
Glenn Jocher b177d01695 updates 2019-04-18 15:01:58 +02:00
Glenn Jocher f4dc0d84e4 updates 2019-04-18 12:27:28 +02:00
Glenn Jocher 8831913f10 updates 2019-04-18 12:21:39 +02:00
Glenn Jocher 5f21139623 updates 2019-04-17 19:06:15 +02:00
Glenn Jocher 2ed0de9785 updates 2019-04-17 19:04:01 +02:00
Glenn Jocher 319c0988cc updates 2019-04-17 18:48:08 +02:00
Glenn Jocher 5fe0346176 updates 2019-04-17 18:40:12 +02:00
Glenn Jocher 7787090165 updates 2019-04-17 18:33:16 +02:00
Glenn Jocher 27ca52c9ee updates 2019-04-17 18:22:40 +02:00
Glenn Jocher ddab3802eb updates 2019-04-17 17:59:01 +02:00
Glenn Jocher 46c55ac3bd updates 2019-04-17 17:51:39 +02:00
Glenn Jocher 663e06f4f9 updates 2019-04-17 17:49:00 +02:00
Glenn Jocher bf966d177f updates 2019-04-17 17:42:17 +02:00
Glenn Jocher 4107315afe updates 2019-04-17 17:41:05 +02:00
Glenn Jocher c2809622c1 updates 2019-04-17 17:35:00 +02:00
Glenn Jocher 5d467c8ac4 updates 2019-04-17 17:31:40 +02:00
Glenn Jocher fb88cb0609 updates 2019-04-17 17:29:23 +02:00
Glenn Jocher f380c7abd2 updates 2019-04-17 17:27:51 +02:00
Glenn Jocher 5f04b93b42 updates 2019-04-17 16:15:08 +02:00
Glenn Jocher 0b8a28e3dd updates 2019-04-17 15:52:51 +02:00
Glenn Jocher b04ea48153 updates 2019-04-17 13:34:17 +02:00
Glenn Jocher 582becc4bf updates 2019-04-17 13:30:28 +02:00
Glenn Jocher 447a3d923d updates 2019-04-17 12:42:10 +02:00
Glenn Jocher 9990e72e6d updates 2019-04-17 00:02:24 +02:00
Glenn Jocher f2e12f6266 updates 2019-04-16 23:47:25 +02:00
Glenn Jocher 54ebb2e593 pin_memory=True 2019-04-15 19:25:36 +02:00
Glenn Jocher e3f0b0248c updates 2019-04-15 18:48:18 +02:00
Glenn Jocher a06abce40b updates 2019-04-15 18:19:45 +02:00
Glenn Jocher 1191dee71b updates 2019-04-15 13:55:52 +02:00
Glenn Jocher 3c6b168a0a updates 2019-04-14 23:22:35 +02:00
Glenn Jocher 09949cdafa updates 2019-04-14 16:00:04 +02:00
Glenn Jocher 52464f5a06 updates 2019-04-13 20:40:49 +02:00
Glenn Jocher 947ee02115 updates 2019-04-13 20:32:29 +02:00
Glenn Jocher f299d83f40 updates 2019-04-13 16:02:45 +02:00
Glenn Jocher 50df252c4b updates 2019-04-12 14:58:19 +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 26b115c306 updates 2019-04-09 12:24:01 +02:00
Glenn Jocher e19b0effb2 updates 2019-04-08 23:45:52 +02:00
Glenn Jocher a34a760d0f updates 2019-04-05 16:26:42 +02:00
Glenn Jocher 54e43b9ad6 updates 2019-04-05 16:19:51 +02:00
Glenn Jocher 65eccee4ef updates 2019-04-05 16:17:15 +02:00
Glenn Jocher 2fab66607c updates 2019-04-05 16:08:18 +02:00
Glenn Jocher 5e79810e69 updates 2019-04-05 15:54:59 +02:00
Glenn Jocher fe896c1792 updates 2019-04-05 15:43:41 +02:00
Glenn Jocher cb352be02c updates 2019-04-05 15:34:42 +02:00
Glenn Jocher 0aff657e19 updates 2019-04-03 14:54:39 +02:00
Glenn Jocher 5b7325bd06 updates 2019-04-03 14:29:28 +02:00
Glenn Jocher 35d2576cb2 default changed to yolov3-spp 2019-04-03 14:25:31 +02:00
Glenn Jocher 5170cd36b0 updates 2019-04-03 11:31:31 +02:00
Glenn Jocher d79a54a4be updates 2019-04-03 11:07:31 +02:00
Glenn Jocher 9b32d3cb5d updates 2019-04-02 19:19:02 +02:00
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