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 |
|
Glenn Jocher
|
6814e925b5
|
updates
|
2019-02-26 03:12:32 +01:00 |
|
Glenn Jocher
|
41ba6dfd6b
|
updates
|
2019-02-26 03:12:04 +01:00 |
|
Glenn Jocher
|
90a20f93e5
|
updates
|
2019-02-26 02:53:11 +01:00 |
|
Glenn Jocher
|
f541861533
|
updates
|
2019-02-25 13:50:01 +01:00 |
|
Glenn Jocher
|
d2cd49f059
|
updates
|
2019-02-25 13:47:51 +01:00 |
|
Glenn Jocher
|
3f68a6776a
|
updates
|
2019-02-21 20:45:53 +01:00 |
|
Glenn Jocher
|
a92b6d4d32
|
updates
|
2019-02-20 23:21:42 +01:00 |
|
Glenn Jocher
|
ead4af98b0
|
updates
|
2019-02-20 15:11:55 +01:00 |
|
Glenn Jocher
|
315fe6ec14
|
xy and wh losses respectively merged
|
2019-02-20 12:50:40 +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
|
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 |