Commit Graph

Select branches

Hide Pull Requests

G0_raw_decode_output

G0_yolov5s6_raw

G10_decoupled_head_light_FGEM_safe

G11_cls_head_enhance_Lightweight

G12_cls_head_enhance_and_ECA

G13_decoupled_head_ECA

G1_backbone

G2_neck

G2_neck_2

G3_head

G4_cls_loss_gain

G5_focal_loss

G6_decoupled_head

G7_RawHead_add_ECA

G8_decoupled_head_light

G9_decoupled_head_light_FGEM

M0_yolov5_raw

master

Mono Color

• 2daa412575 train with multi-gpu half test bug fix #99 Glenn Jocher 2020-06-16 16:28:25 -07:00
• d1e5716610 train with multi-gpu half test bug fix #99 Glenn Jocher 2020-06-16 16:22:58 -07:00
• 1dfc28527f AutoAnchor BPR to 4 significant figures Glenn Jocher 2020-06-16 16:19:58 -07:00
• 8db51c7002 tb_writer bug fix Glenn Jocher 2020-06-16 16:05:28 -07:00
• 4052603e1f AutoAnchor update - improvement check Glenn Jocher 2020-06-16 15:54:20 -07:00
• 5f2eeba233 remove old print statements Alex Stoken 2020-06-16 17:09:39 -05:00
• afe1df385b dist.destroy_process_group() bug fix Glenn Jocher 2020-06-16 15:08:14 -07:00
• 72d06147b2 Multi-GPU disabled in test.py Glenn Jocher 2020-06-16 14:42:30 -07:00
• 333f678b37 add update default hyp dict with provided yaml Alex Stoken 2020-06-16 16:36:20 -05:00
• a448c3bcd7 add logic for resuming and getting hyp for resume run Alex Stoken 2020-06-16 16:30:12 -05:00
• bafbc65ee3 AutoAnchor bug fix Glenn Jocher 2020-06-16 14:26:02 -07:00
• 25e51bcec7 add util function to get most recent last.pt file added logic in train.py __main__ to handle resuming from a run Alex Stoken 2020-06-16 15:50:27 -05:00
• 490f1e7b9c add save_dir arg to plot_lr_scheduler, default to current dir. Uncomment plot_lr_scheduler in train() and pass log_dir as save location Alex Stoken 2020-06-16 15:13:03 -05:00
• 57a0ae3350 AutoAnchor implementation Glenn Jocher 2020-06-16 13:12:22 -07:00
• 4418809cf5 change weights dir (wdir) to be unique to each run, under log_dir Alex Stoken 2020-06-16 15:09:51 -05:00
• d9f446cd81 add save yaml of opt and hyp to tensorboard log_dir in train() Alex Stoken 2020-06-16 15:06:13 -05:00
• a85e6d0fc0 add parser arg for hyp yaml file Alex Stoken 2020-06-16 14:53:32 -05:00
• 95c46f7245 Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library Glenn Jocher 2020-06-16 11:52:36 -07:00
• 5a50491fa1 check_anchors bug fix Glenn Jocher 2020-06-16 10:36:35 -07:00
• 05b8ee5ca4 check_anchors() bug fix #102 Glenn Jocher 2020-06-16 10:34:16 -07:00
• 7c6fd4d670 --verbose bug fix #84 Glenn Jocher 2020-06-16 10:22:22 -07:00
• ec81c7b5f2 check_anchors() bug fix #90 Glenn Jocher 2020-06-16 10:14:04 -07:00
• e1e3399286 ONNX export bug fix #93 Glenn Jocher 2020-06-16 09:59:42 -07:00
• 8b26e89006 AutoAnchor bug fix #72 Glenn Jocher 2020-06-16 00:53:34 -07:00
• 8fa3724072 kmean_anchors() update Glenn Jocher 2020-06-16 00:40:21 -07:00
• bdd9fee841 update fast mode Glenn Jocher 2020-06-16 00:29:54 -07:00
• 1c0b6236e3 update fast mode Glenn Jocher 2020-06-16 00:11:29 -07:00
• 8b8b792802 There is no need to download extra packages, official bring it with you Lornatang 2020-06-16 08:28:52 +08:00
• 0c4b4b8817 Merge remote-tracking branch 'upstream/master' Lornatang 2020-06-16 08:16:10 +08:00
• 09c1b96136 Merge branch 'fix-reference-bugs' Lornatang 2020-06-16 08:15:10 +08:00
• db2c3acd3a updated testing settings, rebalanced towards FP16 latency Glenn Jocher 2020-06-15 16:55:57 -07:00
• 915b1481fc default check_git_status() to True Glenn Jocher 2020-06-15 16:18:46 -07:00
• 0b514daced FP16 test loss bug fix Glenn Jocher 2020-06-15 15:26:29 -07:00
• a1748a8d6e test during training default to FP16 Glenn Jocher 2020-06-15 15:20:27 -07:00
• bd3e389130 update google_utils.py Glenn Jocher 2020-06-15 14:39:04 -07:00
• 76ca367a01 FP16 to FP32 ckpt load Glenn Jocher 2020-06-15 13:32:45 -07:00
• 6fb5ff014c FP16 to FP32 ckpt load Glenn Jocher 2020-06-15 13:21:34 -07:00
• 14523bb030 FP16 to FP32 ckpt load Glenn Jocher 2020-06-15 13:18:39 -07:00
• c5966abba8 glob search bug fix #77 Glenn Jocher 2020-06-15 12:08:57 -07:00
• ace56eb5b6 Merge pull request #70 from Lornatang/fix-reference-bugs Glenn Jocher 2020-06-15 11:18:17 -07:00
• 54795a4f0a fix problem Lornatang 2020-06-15 16:11:06 +08:00
• 45a17ea40c Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-15 00:44:08 -07:00
• e7c24e04ac half() fix #66 Glenn Jocher 2020-06-15 00:43:57 -07:00
• 6570336284 Merge pull request #65 from Ashing00/master Glenn Jocher 2020-06-15 00:34:54 -07:00
• 62eda5e9c2 add "__init__.py" file in <models> folder ashing 2020-06-15 14:37:07 +08:00
• 111994956e switch default inference to FP16 on GPU Glenn Jocher 2020-06-14 19:25:39 -07:00
• 157aff2854 FP16 inference fix Glenn Jocher 2020-06-14 18:45:20 -07:00
• c6fc4242a2 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-14 18:32:35 -07:00
• 260b1729f0 FP16 inference update Glenn Jocher 2020-06-14 18:32:29 -07:00
• b545ee32f4 Created using Colaboratory Glenn Jocher 2020-06-14 13:29:05 -07:00
• 629d3701fc pyyaml >= 5.3 Glenn Jocher 2020-06-14 12:42:51 -07:00
• 4b0e3745f2 Merge pull request #59 from Lijun-Yu/master Glenn Jocher 2020-06-14 12:16:41 -07:00
• 4f6c0cf4c5 Update hubconf.py Glenn Jocher 2020-06-14 12:15:14 -07:00
• 0dec8ffd66 path fix for torch hub Lijun Yu 2020-06-14 14:47:40 -04:00
• 4dd75b3635 pycocotools git install fix Glenn Jocher 2020-06-14 10:54:00 -07:00
• 9bf35e7b86 Merge pull request #56 from channeladam/master Glenn Jocher 2020-06-14 10:33:34 -07:00
• 335c607a24 Update datasets.py Glenn Jocher 2020-06-14 10:32:59 -07:00
• 1374581929 Update requirements.txt Glenn Jocher 2020-06-14 10:23:33 -07:00
• f47ca2085c Include yolov5x extra large model weights Adam 2020-06-14 18:19:23 +10:00
• 34bf936ce6 Add more logging for diagnostics Adam 2020-06-14 18:18:57 +10:00
• faa6aeaa20 Fix pip requirements issues Adam 2020-06-14 18:18:31 +10:00
• 93bbf1b87a best possible recall fix Glenn Jocher 2020-06-13 15:25:53 -07:00
• 31f3310029 assert best possible recall > 0.9 before training Glenn Jocher 2020-06-13 15:05:41 -07:00
• 19e68e8a7b Merge pull request #51 from KPLauritzen/b-ensure-weights-dir-when-training Glenn Jocher 2020-06-13 12:54:43 -07:00
• c3d4d321d3 Ensures weights/ dir exists Lauritzen Kasper Primdal 2020-06-13 14:44:23 +02:00
• 099e6f5ebd --img-size stride-multiple verification Glenn Jocher 2020-06-12 22:10:46 -07:00
• 4ba47b1b1b model.yaml updates Glenn Jocher 2020-06-12 15:41:46 -07:00
• 1e2cb6b480 ONNX export explicit cpu map_location Glenn Jocher 2020-06-12 13:54:00 -07:00
• ad71d2d513 update times to end-to-end latency (inference + NMS) Glenn Jocher 2020-06-12 11:42:08 -07:00
• 8f17a96ff0 cleanup Glenn Jocher 2020-06-11 21:55:22 -07:00
• 8f1755290c ONNX export update Glenn Jocher 2020-06-11 21:54:01 -07:00
• 3a5c5328c5 ONNX, BCEBlurWithLogitsLoss, plot_study updates Glenn Jocher 2020-06-11 14:44:09 -07:00
• 18b4da91e0 PyTorch Hub update Glenn Jocher 2020-06-10 19:14:47 -07:00
• c8f0694894 PyTorch Hub update Glenn Jocher 2020-06-10 19:13:04 -07:00
• a814720403 PyTorch Hub updates Glenn Jocher 2020-06-10 19:11:11 -07:00
• 1e25775c7d Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-10 17:09:43 -07:00
• 0c7a0a08e9 update greetings.yml Glenn Jocher 2020-06-10 17:09:37 -07:00
• f3daa50a32 Update README.md Glenn Jocher 2020-06-10 17:07:25 -07:00
• 46e3cadf28 Update README.md Glenn Jocher 2020-06-10 16:22:18 -07:00
• a4ae9a6850 Update README.md Glenn Jocher 2020-06-10 15:56:03 -07:00
• bb56ed4d05 Merge pull request #35 from pcgeek86/patch-1 Glenn Jocher 2020-06-10 14:42:50 -07:00
• ac93e9c811 Fix Google Drive URI for YOLOv5 weights Trevor Sullivan 2020-06-10 14:26:10 -07:00
• 9b62f9962e bug fix #30 Glenn Jocher 2020-06-10 00:44:34 -07:00
• c14368d768 new CSP model release Glenn Jocher 2020-06-09 14:31:04 -07:00
• 854ccb9a74 _print_weights() Glenn Jocher 2020-06-09 10:51:47 -07:00
• 5bee686649 model definition update Glenn Jocher 2020-06-08 23:26:56 -07:00
• 715cb08b10 module updates Glenn Jocher 2020-06-08 22:15:16 -07:00
• ff02ae0869 module updates Glenn Jocher 2020-06-08 22:13:01 -07:00
• d187459dd4 study changes Glenn Jocher 2020-06-08 10:42:33 -07:00
• 1f698af6f1 create_backbone with SourceChangeWarning update Glenn Jocher 2020-06-07 15:45:05 -07:00
• 546f973562 model fuse Glenn Jocher 2020-06-07 14:47:48 -07:00
• 8fe299f179 model fuse Glenn Jocher 2020-06-07 14:10:30 -07:00
• c672bef10f model fuse Glenn Jocher 2020-06-07 14:07:22 -07:00
• 12b0c046d5 model fusion and onnx export Glenn Jocher 2020-06-07 13:42:33 -07:00
• b810b21299 augmented inference Glenn Jocher 2020-06-05 21:14:15 -07:00
• d5d160449c updates Glenn Jocher 2020-06-05 21:03:17 -07:00
• c12023f145 updates Glenn Jocher 2020-06-05 20:50:43 -07:00
• 5c470d235e multi-gpu test bug fix #15 Glenn Jocher 2020-06-05 20:36:29 -07:00
• 67d5e65c77 profiling update Glenn Jocher 2020-06-05 20:22:39 -07:00
• 2703ac73e8 profiling update Glenn Jocher 2020-06-05 20:22:17 -07:00