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

• a9dc0c2c29 update check_img_size() for model strides Glenn Jocher 2020-06-30 15:25:38 -07:00
• b9da3ea46a Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-30 14:08:14 -07:00
• 3b16c865f0 assert --epochs Glenn Jocher 2020-06-30 14:08:08 -07:00
• cdb9bde181 Unify the check point of single and multi GPU yxNONG 2020-06-30 19:06:28 +08:00
• bfd51f62f8 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-29 17:10:41 -07:00
• 9a9333d245 add replicate() to datasets.py Glenn Jocher 2020-06-29 17:10:33 -07:00
• 615d6d0cfa FROM nvcr.io/nvidia/pytorch:20.06-py3 Glenn Jocher 2020-06-29 14:00:13 -07:00
• e8bf385795 Merge pull request #223 from ArmyOfRobots/master Glenn Jocher 2020-06-29 13:39:53 -07:00
• b203c9b7ff update train.py incompatible model message fix #222 Glenn Jocher 2020-06-29 12:45:25 -07:00
• 01895f3924 Merge pull request #228 from topiaruss/patch-1 Glenn Jocher 2020-06-29 10:08:02 -07:00
• 0d5065d3b9 Remove non-ascii that breaks pdb single-stepping Russ Ferriday 2020-06-29 12:56:40 +01:00
• 0bc80e1e57 onnx_export.py bug fix #224 Glenn Jocher 2020-06-28 17:07:03 -07:00
• ec8554a39b FROM nvcr.io/nvidia/pytorch:20.06-py3 Glenn Jocher 2020-06-28 13:26:31 -07:00
• 685091a3fc Adding torchscript export Jake Poznanski 2020-06-28 13:17:05 -07:00
• 37e13f8846 update mosaic border Glenn Jocher 2020-06-27 13:50:15 -07:00
• 93a6765806 update mosaic border Glenn Jocher 2020-06-27 13:02:01 -07:00
• e18e6811dc Merge branch 'master' into advanced_logging Alex Stoken 2020-06-27 10:13:03 -05:00
• 77fb8ee082 refactor dataloader Glenn Jocher 2020-06-26 19:30:09 -07:00
• 22fb2b0c25 refactor dataloader Glenn Jocher 2020-06-26 18:56:13 -07:00
• 97b5186fa0 Merge pull request #206 from ml5ah/utils-fix Glenn Jocher 2020-06-25 19:26:50 -07:00
• 8c5b6220f5 Update detect.py Glenn Jocher 2020-06-25 19:26:23 -07:00
• 496ec33a33 Update detect.py Glenn Jocher 2020-06-25 19:19:15 -07:00
• 68f63616b3 Fix after suggestion Anish Hiranandani 2020-06-25 18:23:36 -07:00
• 256a3e89d2 small dataset bug fix #140 Glenn Jocher 2020-06-25 17:52:56 -07:00
• d994ed25f1 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-25 15:09:12 -07:00
• e670a3353b update test.py for FP16 testing during training Glenn Jocher 2020-06-25 15:09:04 -07:00
• c910ed397c Update README.md Glenn Jocher 2020-06-25 14:27:45 -07:00
• 8669f4561c Update README.md Glenn Jocher 2020-06-25 12:26:30 -07:00
• 8f7e11dcef Update README.md Glenn Jocher 2020-06-25 11:53:13 -07:00
• b2d4307848 Update README.md Glenn Jocher 2020-06-25 10:59:42 -07:00
• b50fdf16af model.names multi-GPU bug fix #94 Glenn Jocher 2020-06-24 22:22:13 -07:00
• 9a9c4f1259 update yolo.py default --cfg Glenn Jocher 2020-06-24 18:00:43 -07:00
• de191655e4 Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor Alex Stoken 2020-06-24 17:21:54 -05:00
• 2d396bea00 Fix bug in --help from percent sign in help string Alex Stoken 2020-06-24 16:57:12 -05:00
• 227aa735dc updates Glenn Jocher 2020-06-24 13:20:14 -07:00
• e071b8dd3a update models/common.py for Conv() flexible padding Glenn Jocher 2020-06-24 13:12:23 -07:00
• b8557f87e3 add stride to datasets.py Glenn Jocher 2020-06-24 13:02:27 -07:00
• 3b062254a6 update utils.create_pretrained() Glenn Jocher 2020-06-24 12:53:43 -07:00
• d18adf8df9 Add files via upload ChristopherSTAN 2020-06-24 15:53:34 -04:00
• 0825cb7fd8 create models/hub Glenn Jocher 2020-06-24 11:58:23 -07:00
• b1b3634145 update onnx_export.py #134 Glenn Jocher 2020-06-24 11:00:03 -07:00
• 8699c319ca update yolo.py Glenn Jocher 2020-06-24 10:42:01 -07:00
• d1ca6f231d Delete equip_hyp.yaml Alex Stoken 2020-06-24 11:22:12 -05:00
• 611aacf1bf Turn opt into dictionary before sending it to yaml Alex Stoken 2020-06-24 10:49:08 -05:00
• bc4ef4861b Default optimizer SGD Alex Stoken 2020-06-24 10:07:43 -05:00
• 7abf202cad Mode all optimizer settings to 'hyp.yaml', integrate proper momentum with Adam optimizer Alex Stoken 2020-06-24 10:03:21 -05:00
• 7edbf6570e Fix help message for cfg files Alex Stoken 2020-06-24 09:45:57 -05:00
• d64ad0fbf3 Remove --resume functionality and related checks/logic. Alex Stoken 2020-06-24 09:17:27 -05:00
• ccf0af18b1 Revert coco128.yaml to initial commit Alex Stoken 2020-06-24 09:13:41 -05:00
• f9ae460eec update README.md Glenn Jocher 2020-06-23 20:12:23 -07:00
• ec72eea62b Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-23 10:53:48 -07:00
• 53294f28b4 update yolo.py Glenn Jocher 2020-06-23 10:53:42 -07:00
• 6cea1bff60 Update README.md Glenn Jocher 2020-06-23 10:11:34 -07:00
• a63e1c9b3c study update Glenn Jocher 2020-06-22 23:23:50 -07:00
• 2a550333c0 update README.md Glenn Jocher 2020-06-22 23:20:37 -07:00
• db898131f9 study update Glenn Jocher 2020-06-22 23:07:22 -07:00
• 6c1b87a42e update google_utils import Glenn Jocher 2020-06-22 23:00:23 -07:00
• 7306dadf78 Update README.md Glenn Jocher 2020-06-22 15:16:10 -07:00
• 3804578b78 Update README.md Glenn Jocher 2020-06-22 14:33:47 -07:00
• 69ff486693 Update README.md Glenn Jocher 2020-06-22 14:32:05 -07:00
• 364fcfd7db PANet update Glenn Jocher 2020-06-22 14:27:17 -07:00
• ef58dac963 update Dockerfile Glenn Jocher 2020-06-22 10:24:56 -07:00
• bbd12c7cfe detect.py fast bug fix Glenn Jocher 2020-06-22 10:14:58 -07:00
• fc171e2678 check_anchor_order() update Glenn Jocher 2020-06-21 14:08:59 -07:00
• 1f1917ef56 remove fast, add merge Glenn Jocher 2020-06-21 13:37:11 -07:00
• e572bb0803 Add plot_results save location to log_dir Alex Stoken 2020-06-21 09:36:28 -05:00
• 24dd150fbd update torch>=1.4 Glenn Jocher 2020-06-20 11:55:39 -07:00
• ba340ed72d Merge pull request #143 from Lornatang/Fix-multi-GPU-training-errors Glenn Jocher 2020-06-20 11:10:40 -07:00
• 024a42f4fd datasets.py vid_formats updated Glenn Jocher 2020-06-20 09:53:37 -07:00
• 5c2cd711df vid_formats bug fix #146 Glenn Jocher 2020-06-20 09:26:07 -07:00
• 899f1d4bde Fix DDP bug in single process multiple device use cases Lornatang 2020-06-20 13:00:03 +08:00
• 0a09882e4f Update issue templates Glenn Jocher 2020-06-19 16:46:21 -07:00
• cdf1eac9f7 Merge pull request #107 from Lornatang/fix-reference-bugs Glenn Jocher 2020-06-19 14:51:41 -07:00
• 458c4d0e0b Update README.md Glenn Jocher 2020-06-19 12:14:06 -07:00
• 34ec87ecf9 Update README.md Glenn Jocher 2020-06-19 12:13:15 -07:00
• d4c6674c98 speeds updated from FP32 to FP16 Glenn Jocher 2020-06-19 12:05:10 -07:00
• e249792c2f update onnx_export.py usage Glenn Jocher 2020-06-19 10:15:08 -07:00
• 80b82e8bde update yolo.py Glenn Jocher 2020-06-18 19:33:02 -07:00
• 245bd1d242 update README.md Glenn Jocher 2020-06-18 12:46:30 -07:00
• a3ee0a8954 check_img_size() update Glenn Jocher 2020-06-18 11:46:04 -07:00
• 10145833c3 strip_optimizer() default to save FP16 model Glenn Jocher 2020-06-18 00:16:47 -07:00
• cce95e744d backbone as FP16, save default to FP32 Glenn Jocher 2020-06-18 00:13:18 -07:00
• d9b64c27c2 save ckpt in FP16 #119 Glenn Jocher 2020-06-17 22:34:13 -07:00
• 9fdb0fbacf AutoAnchor bug fix # 117 Glenn Jocher 2020-06-17 19:51:15 -07:00
• c8152c81a6 Syntax fixes Alex Stoken 2020-06-17 16:32:13 -05:00
• 9b7386f603 Add save_dir arg to test.test, use arg as location for saving batch jpgs Alex Stoken 2020-06-17 16:08:46 -05:00
• 945307beba Add save_dir to plot_lr_scheduler and plot_labels Set save_dir = log_dir in train.py Alex Stoken 2020-06-17 16:03:18 -05:00
• 3b2b330872 Move results.txt from weights/ to log_dir Alex Stoken 2020-06-17 15:55:45 -05:00
• 8b6f5826bc update README.md Glenn Jocher 2020-06-17 12:58:01 -07:00
• 2bfd242fa1 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-17 12:57:03 -07:00
• b7ac446d62 update README.md Glenn Jocher 2020-06-17 12:56:47 -07:00
• ade023cff2 Fix hyp file read in and dict update. Add example of hyp yaml Alex Stoken 2020-06-17 10:59:20 -05:00
• 3263a204ea Fix get_latest_run to search 'search_dir' recursivly Alex Stoken 2020-06-17 10:34:37 -05:00
• d34291733b Fix get_latest_run() to search inside 'weights' subfolders Alex Stoken 2020-06-17 10:26:55 -05:00
• 91de3381e1 Merge pull request #112 from imprashr/patch-1 Glenn Jocher 2020-06-17 08:23:35 -07:00
• 0071b69994 onnx should be 1.6.0 Prashant Rai 2020-06-17 14:15:27 +05:30
• 2368603484 fix refrence bug Lornatang 2020-06-17 09:56:26 +08:00
• a557b7d924 check_img_size() update Glenn Jocher 2020-06-16 17:48:01 -07:00
• 034609414e AutoAnchor fitness to screen Glenn Jocher 2020-06-16 17:08:45 -07:00
• ef75f7313d AutoAnchor n, imgsz bug fix Glenn Jocher 2020-06-16 16:48:01 -07:00