Commit Graph

970 Commits

Author SHA1 Message Date
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
Glenn Jocher 888cad1e31 updates 2020-02-02 23:35:03 -08:00
Glenn Jocher 785bfec286 updates 2020-02-02 09:19:44 -08:00
Glenn Jocher f7772c791d updates 2020-01-31 09:27:40 -08:00
Glenn Jocher 189c7044fb updates 2020-01-31 09:00:45 -08:00
LinCoce 0c7af1a4d2
fusedconv bug fix, https://github.com/ultralytics/yolov3/issues/807 (#818)
Looks good. Thanks for catching the bug @LinCoce!
2020-01-30 21:58:26 -08:00
Glenn Jocher 6f769081d1 updates 2020-01-30 16:03:34 -08:00
Piotr Skalski 20b0601fa7
change of test batch image format from .jpg to .png, due to matplotlib bug (#817)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-01-30 15:48:26 -08:00
Glenn Jocher 8fac566a87 updates 2020-01-29 14:18:45 -08:00