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 |
|
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
|
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
|
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
|
34bc12d2ad
|
updates
|
2018-11-11 18:58:41 +01:00 |
|
Glenn Jocher
|
6e5da1ce27
|
updates
|
2018-11-05 23:32:36 +01:00 |
|
Glenn Jocher
|
0ae90d0fb7
|
rename /checkpoints to /weights
|
2018-10-27 00:42: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
|
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
|
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
|
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
|
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
|
c0d1aad97e
|
updates
|
2018-08-26 19:38:14 +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 |