Glenn Jocher
2cf23c4aee
add MixConv2d() layer
2020-04-10 18:58:34 -07:00
Glenn Jocher
4bbfda5cde
hist equalization
2020-04-10 17:29:57 -07:00
Glenn Jocher
9bc3a551d9
histogram equalization added to augmentation
2020-04-10 17:24:49 -07:00
Glenn Jocher
398f8eadec
add thr=0.10 to kmean_anchors()
2020-04-10 16:34:32 -07:00
Glenn Jocher
aa8b1098dd
adapt mosaic to img channel count
2020-04-10 16:28:59 -07:00
Glenn Jocher
6736d7d125
swap cv2.INTER_AREA for cv2.INTER_LINEAR
2020-04-10 12:47:07 -07:00
Glenn Jocher
4120ac3aa6
training updates
2020-04-08 21:01:58 -07:00
Glenn Jocher
f54d28ba63
improve assert no labels found
2020-04-08 09:57:59 -07:00
Glenn Jocher
6c5ecaf805
remove label loading during training
2020-04-07 16:57:22 -07:00
Glenn Jocher
b9b14bef59
scale_img() bug fix
2020-04-07 13:35:47 -07:00
Glenn Jocher
067ee264c0
scale_img() bug fix
2020-04-07 12:59:52 -07:00
Glenn Jocher
68f58f4dec
scale_img() update
2020-04-07 12:51:52 -07:00
Glenn Jocher
26fc4fb018
dataloader default color to imagenet mean 114
2020-04-05 18:02:41 -07:00
Glenn Jocher
4da5c6c114
rect padding to 64, mAP increase 42.7 to 42.9
2020-04-05 16:06:27 -07:00
Glenn Jocher
a657345b45
add FeatureConcat() module
2020-04-05 14:47:41 -07:00
Glenn Jocher
a19b1a3b94
line thickness
2020-04-05 11:10:05 -07:00
Glenn Jocher
6203340888
detect.py multi_label default False
2020-04-05 11:05:49 -07:00
Glenn Jocher
00c1fdd805
add MixConv2d() layer
2020-04-03 20:03:44 -07:00
Glenn Jocher
207c6fcff9
merge NMS full matrix
2020-04-02 18:53:40 -07:00
Glenn Jocher
27c7334e81
new layers.py file
2020-04-02 12:22:15 -07:00
Glenn Jocher
4ac60018f6
FLOPS report
2020-04-01 14:05:41 -07:00
Glenn Jocher
300e9a7ad6
merge NMS full matrix
2020-03-31 21:31:09 -07:00
Glenn Jocher
02802e67f2
merge NMS full matrix
2020-03-31 18:18:08 -07:00
Glenn Jocher
f4eecef700
merge NMS speed/memory improvements
2020-03-31 15:37:23 -07:00
Glenn Jocher
108334db29
FLOPs update
2020-03-30 16:04:08 -07:00
Glenn Jocher
105882b3c6
GFLOPs correction
2020-03-30 15:30:53 -07:00
Glenn Jocher
eb151a881e
NMS and test batch_size updates
2020-03-29 20:41:32 -07:00
Glenn Jocher
9c5e76b93d
EMA implemented by default
2020-03-29 13:14:54 -07:00
Glenn Jocher
f9d34587da
Merge NMS update
2020-03-27 13:11:24 -07:00
GoogleWiki
582de735ad
utils.clip_coords doesn't work as expected. ( #961 )
...
* utils.clip_coords doesn't work as expected.
Box coords may be negative or exceed borders.
* Update utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-03-27 13:09:10 -07:00
Glenn Jocher
4a63b24b09
Merge NMS update
2020-03-26 19:50:29 -07:00
Glenn Jocher
01ee0c5e95
Merge NMS update
2020-03-26 18:41:04 -07:00
Glenn Jocher
dad59220f1
speed and comments update
2020-03-26 18:34:20 -07:00
Glenn Jocher
a322fc5d4b
Merge NMS update
2020-03-26 12:48:00 -07:00
Glenn Jocher
171b4129b5
Merge NMS update
2020-03-26 12:33:12 -07:00
Glenn Jocher
eac07f9da3
Merge NMS update
2020-03-26 12:20:01 -07:00
Glenn Jocher
23b34f4db8
merge_batch NMS method
2020-03-25 23:29:33 -07:00
Glenn Jocher
aa0c64b5ac
merge_batch NMS method
2020-03-25 23:24:57 -07:00
Glenn Jocher
83c9cfb7de
FocalLoss() and obj loss speed and stability update
2020-03-19 12:30:37 -07:00
Glenn Jocher
20454990ce
FLOPS report
2020-03-19 12:30:07 -07:00
Glenn Jocher
60c8d194cd
FocalLoss() and obj loss speed and stability update
2020-03-19 11:52:52 -07:00
Glenn Jocher
b3adc896f9
focal and obj loss speed/stability
2020-03-16 21:40:57 -07:00
Glenn Jocher
448c4a6e1f
Remove deprecated --arc architecture options, implement --arc default for all cases
2020-03-16 20:46:25 -07:00
Glenn Jocher
77c6c01970
EMA class updates
2020-03-16 17:51:40 -07:00
Glenn Jocher
1a12667ce1
loss function cleanup
2020-03-16 17:31:37 -07:00
Glenn Jocher
f1208f784e
updated run history
2020-03-16 15:36:56 -07:00
Glenn Jocher
c09fcfc4fe
EMA class updates
2020-03-16 14:18:56 -07:00
Glenn Jocher
c4047000fe
FocalLoss() updated to match TF
2020-03-16 14:03:50 -07:00
Glenn Jocher
07d2f0ad03
test/inference time augmentation
2020-03-15 18:39:54 -07:00
Glenn Jocher
adba66c3a6
EMA class updates
2020-03-14 18:08:48 -07:00
Glenn Jocher
9ce4ec48a7
model.info() method implemented
2020-03-14 16:46:54 -07:00
Glenn Jocher
b89cc396af
EMA class updates
2020-03-14 16:23:14 -07:00
Falak
666ba85ed3
Comment updates on box coordinates ( #852 )
...
* Update utils.py
Reusing function defined above
* Update utils.py
* Reverting change which break bbox coordinate computation
* Update utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-03-14 15:35:59 -07:00
Glenn Jocher
a52c0abf8d
updates
2020-03-13 20:12:54 -07:00
Glenn Jocher
418269d739
FocalLoss() gamma and alpha default values
2020-03-13 16:51:30 -07:00
Glenn Jocher
f30a8706e5
update to coco results82
2020-03-13 12:06:48 -07:00
Glenn Jocher
0de07da612
updates
2020-03-13 11:03:39 -07:00
Glenn Jocher
6aae5aca64
inplace clip_coords() clamp
2020-03-13 10:47:00 -07:00
Glenn Jocher
a278623067
replaced floatn() with round()
2020-03-13 10:06:17 -07:00
Glenn Jocher
8a1f35eac6
updates
2020-03-12 01:09:17 -07:00
Glenn Jocher
6ca8277de2
updates
2020-03-11 21:30:47 -07:00
Glenn Jocher
585064f300
updates
2020-03-10 13:33:14 -07:00
Glenn Jocher
6bd51b75ea
updates
2020-03-09 11:20:22 -07:00
Glenn Jocher
4a90221e79
updates
2020-03-08 16:15:41 -07:00
Glenn Jocher
1d43b2a55a
updates
2020-03-08 16:13:56 -07:00
Glenn Jocher
a4662bf306
updates
2020-03-08 11:53:18 -07:00
Glenn Jocher
feea9c1a65
P and R evaluated at 0.5 score
2020-03-07 10:26:08 -08:00
Glenn Jocher
692b006f4d
updates
2020-03-05 14:20:52 -08:00
Glenn Jocher
1dc1761f45
updates
2020-03-05 10:20:08 -08:00
Glenn Jocher
b8b89a3132
updates
2020-03-05 09:54:41 -08:00
Glenn Jocher
981b452b1d
updates
2020-03-04 13:20:08 -08:00
Glenn Jocher
1430a1e408
updates
2020-03-04 10:26:35 -08:00
Glenn Jocher
35eae3ace9
updates
2020-03-04 09:53:02 -08:00
Glenn Jocher
be01fc357b
updates
2020-03-04 00:22:01 -08:00
Glenn Jocher
f915bf175c
updates
2020-03-04 00:08:18 -08:00
Glenn Jocher
166f8c0e53
updates
2020-03-04 00:07:19 -08:00
Glenn Jocher
308f7c8563
updates
2020-03-03 19:16:13 -08:00
Glenn Jocher
dce753ead4
updates
2020-03-02 14:30:01 -08:00
Glenn Jocher
b3ecfb10bc
updates
2020-02-27 22:50:26 -08:00
Glenn Jocher
3a1ca6454c
updates
2020-02-27 13:00:00 -08:00
Glenn Jocher
6a99e39bd5
updates
2020-02-27 12:57:10 -08:00
Glenn Jocher
de3e539609
updates
2020-02-27 12:49:01 -08:00
Glenn Jocher
f3d3295f90
updates
2020-02-27 12:38:14 -08:00
Glenn Jocher
7e92f70e05
updates
2020-02-27 12:19:06 -08:00
Glenn Jocher
764514e44d
updates
2020-02-26 13:52:33 -08:00
Glenn Jocher
4b720013d1
updates
2020-02-24 12:43:13 -08:00
Glenn Jocher
a3671bde94
updates
2020-02-23 18:33:32 -08:00
Glenn Jocher
bc741f30e8
updates
2020-02-22 18:18:38 -08:00
Glenn Jocher
7608047531
updates
2020-02-22 17:43:11 -08:00
Glenn Jocher
fa8882c98e
updates
2020-02-21 15:11:11 -08:00
Glenn Jocher
afbc2f8d78
updates
2020-02-21 15:10:50 -08:00
Glenn Jocher
6fbab656c8
updates
2020-02-19 17:08:03 -08:00
Glenn Jocher
ddd892dc20
updates
2020-02-18 21:04:58 -08:00
Glenn Jocher
a971b33b74
updates
2020-02-17 17:34:40 -08:00
Glenn Jocher
45ce01f859
updates
2020-02-17 15:28:11 -08:00
Glenn Jocher
e840b7c781
add yolov3-spp-ultralytics.pt
2020-02-16 23:12:07 -08:00
Glenn Jocher
ca4960f7ff
updates
2020-02-08 13:28:47 -08:00
Yonghye Kwon
145ea67a2e
modify h range clip range in hsv augmentation ( #825 )
...
* h range clip range edit in hsv augmentation
h range is [0., 179,]
* Update datasets.py
reduced indexing operations and used inplace clip for hsv. Two clips are used unfortunately (double clip of axis 0), but the overall effect should be improved speed.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-02-07 09:14:55 -08:00
Glenn Jocher
dca80f6f98
updates
2020-02-06 16:13:10 -08:00
Glenn Jocher
ec942bd23c
updates
2020-02-05 20:27:01 -08:00