944 Commits

Author SHA1 Message Date
Glenn Jocher
510eadcfa5 Apex and 'git pull' suggestions 2020-04-15 12:54:56 -07:00
Glenn Jocher
a49ea80218 update initialize_weights() 2020-04-14 15:58:32 -07:00
Glenn Jocher
ac4c90c817 cleanup 2020-04-14 13:08:00 -07:00
Glenn Jocher
763cdd5ae2 detailed image sizes report 2020-04-14 11:51:19 -07:00
Glenn Jocher
029e137bc2 bug fix 2020-04-14 04:34:40 -07:00
Glenn Jocher
1681249588 cleanup 2020-04-14 04:15:53 -07:00
Glenn Jocher
198a5a591d code cleanup 2020-04-14 04:15:05 -07:00
Glenn Jocher
25725c8569 bug fix 2020-04-14 03:13:30 -07:00
Glenn Jocher
835b0da68a new modules and init weights 2020-04-14 01:20:57 -07:00
Glenn Jocher
76fb8d48d4 ng dependence removed from build_targets() 2020-04-13 21:25:03 -07:00
Glenn Jocher
b8574add37 new find_modules() fcn 2020-04-13 17:48:30 -07:00
Glenn Jocher
77e6bdd3c1 FLOPs at 480x640, BN init 2020-04-12 18:44:18 -07:00
Glenn Jocher
efc754a794 add generations arg to kmeans() 2020-04-12 12:49:23 -07:00
Glenn Jocher
a34219a54b pading from (k-1) // 2 to k // 2 2020-04-11 12:18:54 -07:00
Glenn Jocher
b574f765ce add warning to plot_results() 2020-04-11 11:04:10 -07:00
Glenn Jocher
58edfc4a84 kaiming weight init 2020-04-11 10:45:33 -07:00
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