Commit Graph

193 Commits

Author SHA1 Message Date
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