Commit Graph

  • 6c1b87a42e update google_utils import Glenn Jocher 2020-06-22 23:00:23 -0700
  • 7306dadf78
    Update README.md Glenn Jocher 2020-06-22 15:16:10 -0700
  • 3804578b78
    Update README.md Glenn Jocher 2020-06-22 14:33:47 -0700
  • 69ff486693
    Update README.md Glenn Jocher 2020-06-22 14:32:05 -0700
  • 364fcfd7db PANet update Glenn Jocher 2020-06-22 14:27:17 -0700
  • ef58dac963 update Dockerfile Glenn Jocher 2020-06-22 10:24:56 -0700
  • bbd12c7cfe detect.py fast bug fix Glenn Jocher 2020-06-22 10:14:58 -0700
  • fc171e2678 check_anchor_order() update Glenn Jocher 2020-06-21 14:08:59 -0700
  • 1f1917ef56 remove fast, add merge Glenn Jocher 2020-06-21 13:37:11 -0700
  • e572bb0803 Add plot_results save location to log_dir Alex Stoken 2020-06-21 09:36:28 -0500
  • 24dd150fbd update torch>=1.4 Glenn Jocher 2020-06-20 11:55:39 -0700
  • ba340ed72d
    Merge pull request #143 from Lornatang/Fix-multi-GPU-training-errors Glenn Jocher 2020-06-20 11:10:40 -0700
  • 024a42f4fd datasets.py vid_formats updated Glenn Jocher 2020-06-20 09:53:37 -0700
  • 5c2cd711df vid_formats bug fix #146 Glenn Jocher 2020-06-20 09:26:07 -0700
  • 899f1d4bde Fix DDP bug in single process multiple device use cases Lornatang 2020-06-20 13:00:03 +0800
  • 0a09882e4f Update issue templates Glenn Jocher 2020-06-19 16:46:21 -0700
  • cdf1eac9f7
    Merge pull request #107 from Lornatang/fix-reference-bugs Glenn Jocher 2020-06-19 14:51:41 -0700
  • 458c4d0e0b
    Update README.md Glenn Jocher 2020-06-19 12:14:06 -0700
  • 34ec87ecf9
    Update README.md Glenn Jocher 2020-06-19 12:13:15 -0700
  • d4c6674c98 speeds updated from FP32 to FP16 Glenn Jocher 2020-06-19 12:05:10 -0700
  • e249792c2f update onnx_export.py usage Glenn Jocher 2020-06-19 10:15:08 -0700
  • 80b82e8bde update yolo.py Glenn Jocher 2020-06-18 19:33:02 -0700
  • 245bd1d242 update README.md Glenn Jocher 2020-06-18 12:46:30 -0700
  • a3ee0a8954 check_img_size() update Glenn Jocher 2020-06-18 11:46:04 -0700
  • 10145833c3 strip_optimizer() default to save FP16 model Glenn Jocher 2020-06-18 00:16:47 -0700
  • cce95e744d backbone as FP16, save default to FP32 Glenn Jocher 2020-06-18 00:13:18 -0700
  • d9b64c27c2 save ckpt in FP16 #119 Glenn Jocher 2020-06-17 22:34:13 -0700
  • 9fdb0fbacf AutoAnchor bug fix # 117 Glenn Jocher 2020-06-17 19:51:15 -0700
  • c8152c81a6 Syntax fixes Alex Stoken 2020-06-17 16:32:13 -0500
  • 9b7386f603 Add save_dir arg to test.test, use arg as location for saving batch jpgs Alex Stoken 2020-06-17 16:08:46 -0500
  • 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 -0500
  • 3b2b330872 Move results.txt from weights/ to log_dir Alex Stoken 2020-06-17 15:55:45 -0500
  • 8b6f5826bc update README.md Glenn Jocher 2020-06-17 12:58:01 -0700
  • 2bfd242fa1 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-17 12:57:03 -0700
  • b7ac446d62 update README.md Glenn Jocher 2020-06-17 12:56:47 -0700
  • ade023cff2 Fix hyp file read in and dict update. Add example of hyp yaml Alex Stoken 2020-06-17 10:59:20 -0500
  • 3263a204ea Fix get_latest_run to search 'search_dir' recursivly Alex Stoken 2020-06-17 10:34:37 -0500
  • d34291733b Fix get_latest_run() to search inside 'weights' subfolders Alex Stoken 2020-06-17 10:26:55 -0500
  • 91de3381e1
    Merge pull request #112 from imprashr/patch-1 Glenn Jocher 2020-06-17 08:23:35 -0700
  • 0071b69994
    onnx should be 1.6.0 Prashant Rai 2020-06-17 14:15:27 +0530
  • 2368603484 fix refrence bug Lornatang 2020-06-17 09:56:26 +0800
  • a557b7d924 check_img_size() update Glenn Jocher 2020-06-16 17:48:01 -0700
  • 034609414e AutoAnchor fitness to screen Glenn Jocher 2020-06-16 17:08:45 -0700
  • ef75f7313d AutoAnchor n, imgsz bug fix Glenn Jocher 2020-06-16 16:48:01 -0700
  • 2daa412575 train with multi-gpu half test bug fix #99 Glenn Jocher 2020-06-16 16:28:25 -0700
  • d1e5716610 train with multi-gpu half test bug fix #99 Glenn Jocher 2020-06-16 16:22:58 -0700
  • 1dfc28527f AutoAnchor BPR to 4 significant figures Glenn Jocher 2020-06-16 16:19:58 -0700
  • 8db51c7002 tb_writer bug fix Glenn Jocher 2020-06-16 16:05:28 -0700
  • 4052603e1f AutoAnchor update - improvement check Glenn Jocher 2020-06-16 15:54:20 -0700
  • 5f2eeba233 remove old print statements Alex Stoken 2020-06-16 17:09:39 -0500
  • afe1df385b dist.destroy_process_group() bug fix Glenn Jocher 2020-06-16 15:08:14 -0700
  • 72d06147b2 Multi-GPU disabled in test.py Glenn Jocher 2020-06-16 14:42:30 -0700
  • 333f678b37 add update default hyp dict with provided yaml Alex Stoken 2020-06-16 16:36:20 -0500
  • a448c3bcd7 add logic for resuming and getting hyp for resume run Alex Stoken 2020-06-16 16:30:12 -0500
  • bafbc65ee3 AutoAnchor bug fix Glenn Jocher 2020-06-16 14:26:02 -0700
  • 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 -0500
  • 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 -0500
  • 57a0ae3350 AutoAnchor implementation Glenn Jocher 2020-06-16 13:12:22 -0700
  • 4418809cf5 change weights dir (wdir) to be unique to each run, under log_dir Alex Stoken 2020-06-16 15:09:51 -0500
  • d9f446cd81 add save yaml of opt and hyp to tensorboard log_dir in train() Alex Stoken 2020-06-16 15:06:13 -0500
  • a85e6d0fc0 add parser arg for hyp yaml file Alex Stoken 2020-06-16 14:53:32 -0500
  • 95c46f7245
    Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library Glenn Jocher 2020-06-16 11:52:36 -0700
  • 5a50491fa1 check_anchors bug fix Glenn Jocher 2020-06-16 10:36:35 -0700
  • 05b8ee5ca4 check_anchors() bug fix #102 Glenn Jocher 2020-06-16 10:34:16 -0700
  • 7c6fd4d670 --verbose bug fix #84 Glenn Jocher 2020-06-16 10:22:22 -0700
  • ec81c7b5f2 check_anchors() bug fix #90 Glenn Jocher 2020-06-16 10:14:04 -0700
  • e1e3399286 ONNX export bug fix #93 Glenn Jocher 2020-06-16 09:59:42 -0700
  • 8b26e89006 AutoAnchor bug fix #72 Glenn Jocher 2020-06-16 00:53:34 -0700
  • 8fa3724072 kmean_anchors() update Glenn Jocher 2020-06-16 00:40:21 -0700
  • bdd9fee841 update fast mode Glenn Jocher 2020-06-16 00:29:54 -0700
  • 1c0b6236e3 update fast mode Glenn Jocher 2020-06-16 00:11:29 -0700
  • 8b8b792802 There is no need to download extra packages, official bring it with you Lornatang 2020-06-16 08:28:52 +0800
  • 0c4b4b8817 Merge remote-tracking branch 'upstream/master' Lornatang 2020-06-16 08:16:10 +0800
  • 09c1b96136 Merge branch 'fix-reference-bugs' Lornatang 2020-06-16 08:15:10 +0800
  • db2c3acd3a updated testing settings, rebalanced towards FP16 latency Glenn Jocher 2020-06-15 16:55:57 -0700
  • 915b1481fc default check_git_status() to True Glenn Jocher 2020-06-15 16:18:46 -0700
  • 0b514daced FP16 test loss bug fix Glenn Jocher 2020-06-15 15:26:29 -0700
  • a1748a8d6e test during training default to FP16 Glenn Jocher 2020-06-15 15:20:27 -0700
  • bd3e389130 update google_utils.py Glenn Jocher 2020-06-15 14:39:04 -0700
  • 76ca367a01 FP16 to FP32 ckpt load Glenn Jocher 2020-06-15 13:32:45 -0700
  • 6fb5ff014c FP16 to FP32 ckpt load Glenn Jocher 2020-06-15 13:21:34 -0700
  • 14523bb030 FP16 to FP32 ckpt load Glenn Jocher 2020-06-15 13:18:39 -0700
  • c5966abba8 glob search bug fix #77 Glenn Jocher 2020-06-15 12:08:57 -0700
  • ace56eb5b6
    Merge pull request #70 from Lornatang/fix-reference-bugs Glenn Jocher 2020-06-15 11:18:17 -0700
  • 54795a4f0a fix problem Lornatang 2020-06-15 16:11:06 +0800
  • 45a17ea40c Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-15 00:44:08 -0700
  • e7c24e04ac half() fix #66 Glenn Jocher 2020-06-15 00:43:57 -0700
  • 6570336284
    Merge pull request #65 from Ashing00/master Glenn Jocher 2020-06-15 00:34:54 -0700
  • 62eda5e9c2 add "__init__.py" file in <models> folder ashing 2020-06-15 14:37:07 +0800
  • 111994956e switch default inference to FP16 on GPU Glenn Jocher 2020-06-14 19:25:39 -0700
  • 157aff2854 FP16 inference fix Glenn Jocher 2020-06-14 18:45:20 -0700
  • c6fc4242a2 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-06-14 18:32:35 -0700
  • 260b1729f0 FP16 inference update Glenn Jocher 2020-06-14 18:32:29 -0700
  • b545ee32f4 Created using Colaboratory Glenn Jocher 2020-06-14 13:29:05 -0700
  • 629d3701fc pyyaml >= 5.3 Glenn Jocher 2020-06-14 12:42:51 -0700
  • 4b0e3745f2
    Merge pull request #59 from Lijun-Yu/master Glenn Jocher 2020-06-14 12:16:41 -0700
  • 4f6c0cf4c5
    Update hubconf.py Glenn Jocher 2020-06-14 12:15:14 -0700
  • 0dec8ffd66 path fix for torch hub Lijun Yu 2020-06-14 14:47:40 -0400
  • 4dd75b3635 pycocotools git install fix Glenn Jocher 2020-06-14 10:54:00 -0700
  • 9bf35e7b86
    Merge pull request #56 from channeladam/master Glenn Jocher 2020-06-14 10:33:34 -0700