Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
244 Commits 1 Branch 0 Tags
655895a8382fcafd0e346794f401c081f57f2dc4
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher 655895a838 module updates 2020-07-03 13:46:12 -07:00
Glenn Jocher df224a0d8f EMA bug fix #279 2020-07-03 11:56:14 -07:00
Glenn Jocher a9d20eba3e export.py update 2020-07-03 11:50:59 -07:00
Glenn Jocher 569fbc63c8 export.py update 2020-07-03 11:29:53 -07:00
Glenn Jocher 3bdea3f697 strip_optimizer() bug fix #253 2020-07-02 21:24:26 -07:00
Glenn Jocher 6ca3f35cd4 update .dockerignore 2020-07-02 16:41:03 -07:00
Glenn Jocher e02a189a3a Merge pull request #245 from yxNONG/patch-2 
Unify the check point of single and multi GPU
 2020-07-02 12:05:46 -07:00
Glenn Jocher f02481c73a Update torch_utils.py 2020-07-02 12:03:45 -07:00
Glenn Jocher fc7c42723d Update utils.py 2020-07-02 12:01:43 -07:00
Glenn Jocher 597ed4ce63 Update train.py 2020-07-02 12:00:55 -07:00
Glenn Jocher 13f69777a6 typo fix 2020-07-02 09:26:03 -07:00
Glenn Jocher a03adb5ab8 Merge remote-tracking branch 'origin/master' 2020-07-01 19:16:06 -07:00
Glenn Jocher 1fca7a7f24 autopad() update in common.py 2020-07-01 19:15:59 -07:00
Glenn Jocher 5323ad224d export.py update 2020-07-01 16:15:25 -07:00
Glenn Jocher a62a1c2c67 export.py update 2020-07-01 16:14:49 -07:00
Glenn Jocher f1d67f4110 update export.py 2020-07-01 15:46:15 -07:00
Glenn Jocher e74ccb2985 Merge remote-tracking branch 'origin/master' 2020-07-01 11:44:55 -07:00
Glenn Jocher b5659d1195 module updates 2020-07-01 11:44:49 -07:00
Glenn Jocher dfd63de20a Merge pull request #255 from edurenye/ignore_torchscript 
Add torchscript files to gitignore
 2020-07-01 10:10:11 -07:00
Glenn Jocher 86784cfdbf --resume bug fix #252 2020-06-30 21:43:53 -07:00
Glenn Jocher 0a08375a8a Merge remote-tracking branch 'origin/master' 2020-06-30 16:42:16 -07:00
Glenn Jocher 4e62eb90be check_img_size() int() update 2020-06-30 16:41:26 -07:00
Glenn Jocher 5c1c73ecda Merge remote-tracking branch 'origin/master' 2020-06-30 16:16:36 -07:00
Glenn Jocher ad4c22cbfe --resume bug fix #187 2020-06-30 16:16:29 -07:00
Glenn Jocher 2b7a11300e Merge remote-tracking branch 'origin/master' 2020-06-30 15:25:45 -07:00
Glenn Jocher a9dc0c2c29 update check_img_size() for model strides 2020-06-30 15:25:38 -07:00
Glenn Jocher b9da3ea46a Merge remote-tracking branch 'origin/master' 2020-06-30 14:08:14 -07:00
Glenn Jocher 3b16c865f0 assert --epochs 2020-06-30 14:08:08 -07:00
Glenn Jocher bfd51f62f8 Merge remote-tracking branch 'origin/master' 2020-06-29 17:10:41 -07:00
Glenn Jocher 9a9333d245 add replicate() to datasets.py 2020-06-29 17:10:33 -07:00
Glenn Jocher 615d6d0cfa FROM nvcr.io/nvidia/pytorch:20.06-py3 2020-06-29 14:00:13 -07:00
Glenn Jocher e8bf385795 Merge pull request #223 from ArmyOfRobots/master 
Adding torchscript export
 2020-06-29 13:39:53 -07:00
Glenn Jocher b203c9b7ff update train.py incompatible model message fix #222 2020-06-29 12:45:25 -07:00
Glenn Jocher 01895f3924 Merge pull request #228 from topiaruss/patch-1 
Remove non-ascii that breaks pdb single-stepping
 2020-06-29 10:08:02 -07:00
Glenn Jocher 0bc80e1e57 onnx_export.py bug fix #224 2020-06-28 17:07:03 -07:00
Glenn Jocher ec8554a39b FROM nvcr.io/nvidia/pytorch:20.06-py3 2020-06-28 13:26:31 -07:00
Glenn Jocher 37e13f8846 update mosaic border 2020-06-27 13:50:15 -07:00
Glenn Jocher 93a6765806 update mosaic border 2020-06-27 13:02:01 -07:00
Glenn Jocher 77fb8ee082 refactor dataloader 2020-06-26 19:30:09 -07:00
Glenn Jocher 22fb2b0c25 refactor dataloader 2020-06-26 18:56:13 -07:00
Glenn Jocher 97b5186fa0 Merge pull request #206 from ml5ah/utils-fix 
Fix after suggestion
 2020-06-25 19:26:50 -07:00
Glenn Jocher 8c5b6220f5 Update detect.py 2020-06-25 19:26:23 -07:00
Glenn Jocher 496ec33a33 Update detect.py 
Added some recent updates that were missing, and updated the filename with an if else.
 2020-06-25 19:19:15 -07:00
Glenn Jocher 256a3e89d2 small dataset bug fix #140 2020-06-25 17:52:56 -07:00
Glenn Jocher d994ed25f1 Merge remote-tracking branch 'origin/master' 2020-06-25 15:09:12 -07:00
Glenn Jocher e670a3353b update test.py for FP16 testing during training 2020-06-25 15:09:04 -07:00
Glenn Jocher c910ed397c Update README.md 2020-06-25 14:27:45 -07:00
Glenn Jocher 8669f4561c Update README.md 2020-06-25 12:26:30 -07:00
Glenn Jocher 8f7e11dcef Update README.md 2020-06-25 11:53:13 -07:00
Glenn Jocher b2d4307848 Update README.md 2020-06-25 10:59:42 -07:00
Glenn Jocher b50fdf16af model.names multi-GPU bug fix #94 2020-06-24 22:22:13 -07:00
Glenn Jocher 9a9c4f1259 update yolo.py default --cfg 2020-06-24 18:00:43 -07:00
Glenn Jocher 227aa735dc updates 2020-06-24 13:20:14 -07:00
Glenn Jocher e071b8dd3a update models/common.py for Conv() flexible padding 2020-06-24 13:12:23 -07:00
Glenn Jocher b8557f87e3 add stride to datasets.py 2020-06-24 13:02:27 -07:00
Glenn Jocher 3b062254a6 update utils.create_pretrained() 2020-06-24 12:53:43 -07:00
Glenn Jocher 0825cb7fd8 create models/hub 2020-06-24 11:58:23 -07:00
Glenn Jocher b1b3634145 update onnx_export.py #134 2020-06-24 11:00:03 -07:00
Glenn Jocher 8699c319ca update yolo.py 2020-06-24 10:42:01 -07:00
Glenn Jocher f9ae460eec update README.md 2020-06-23 20:12:23 -07:00
Glenn Jocher ec72eea62b Merge remote-tracking branch 'origin/master' 2020-06-23 10:53:48 -07:00
Glenn Jocher 53294f28b4 update yolo.py 2020-06-23 10:53:42 -07:00
Glenn Jocher 6cea1bff60 Update README.md 2020-06-23 10:11:34 -07:00
Glenn Jocher a63e1c9b3c study update 2020-06-22 23:23:50 -07:00
Glenn Jocher 2a550333c0 update README.md 2020-06-22 23:20:37 -07:00
Glenn Jocher db898131f9 study update 2020-06-22 23:07:22 -07:00
Glenn Jocher 6c1b87a42e update google_utils import 2020-06-22 23:00:23 -07:00
Glenn Jocher 7306dadf78 Update README.md 2020-06-22 15:16:10 -07:00
Glenn Jocher 3804578b78 Update README.md 2020-06-22 14:33:47 -07:00
Glenn Jocher 69ff486693 Update README.md 2020-06-22 14:32:05 -07:00
Glenn Jocher 364fcfd7db PANet update 2020-06-22 14:27:17 -07:00
Glenn Jocher ef58dac963 update Dockerfile 2020-06-22 10:24:56 -07:00
Glenn Jocher bbd12c7cfe detect.py fast bug fix 2020-06-22 10:14:58 -07:00
Glenn Jocher fc171e2678 check_anchor_order() update 2020-06-21 14:08:59 -07:00
Glenn Jocher 1f1917ef56 remove fast, add merge 2020-06-21 13:37:11 -07:00
Glenn Jocher 24dd150fbd update torch>=1.4 2020-06-20 11:55:39 -07:00
Glenn Jocher ba340ed72d Merge pull request #143 from Lornatang/Fix-multi-GPU-training-errors 
Fix DDP bug in single process multiple device use cases
 2020-06-20 11:10:40 -07:00
Glenn Jocher 024a42f4fd datasets.py vid_formats updated 2020-06-20 09:53:37 -07:00
Glenn Jocher 5c2cd711df vid_formats bug fix #146 2020-06-20 09:26:07 -07:00
Glenn Jocher 0a09882e4f Update issue templates 2020-06-19 16:46:21 -07:00
Glenn Jocher cdf1eac9f7 Merge pull request #107 from Lornatang/fix-reference-bugs 
fix refrence bug
 2020-06-19 14:51:41 -07:00
Glenn Jocher 458c4d0e0b Update README.md 2020-06-19 12:14:06 -07:00
Glenn Jocher 34ec87ecf9 Update README.md 2020-06-19 12:13:15 -07:00
Glenn Jocher d4c6674c98 speeds updated from FP32 to FP16 2020-06-19 12:05:10 -07:00
Glenn Jocher e249792c2f update onnx_export.py usage 2020-06-19 10:15:08 -07:00
Glenn Jocher 80b82e8bde update yolo.py 2020-06-18 19:33:02 -07:00
Glenn Jocher 245bd1d242 update README.md 2020-06-18 12:46:30 -07:00
Glenn Jocher a3ee0a8954 check_img_size() update 2020-06-18 11:46:04 -07:00
Glenn Jocher 10145833c3 strip_optimizer() default to save FP16 model 2020-06-18 00:16:47 -07:00
Glenn Jocher cce95e744d backbone as FP16, save default to FP32 2020-06-18 00:13:18 -07:00
Glenn Jocher d9b64c27c2 save ckpt in FP16 #119 2020-06-17 22:34:13 -07:00
Glenn Jocher 9fdb0fbacf AutoAnchor bug fix # 117 2020-06-17 19:51:15 -07:00
Glenn Jocher 8b6f5826bc update README.md 2020-06-17 12:58:01 -07:00
Glenn Jocher 2bfd242fa1 Merge remote-tracking branch 'origin/master' 2020-06-17 12:57:03 -07:00
Glenn Jocher b7ac446d62 update README.md 2020-06-17 12:56:47 -07:00
Glenn Jocher 91de3381e1 Merge pull request #112 from imprashr/patch-1 
onnx should be 1.6.0
 2020-06-17 08:23:35 -07:00
Glenn Jocher a557b7d924 check_img_size() update 2020-06-16 17:48:01 -07:00
Glenn Jocher 034609414e AutoAnchor fitness to screen 2020-06-16 17:08:45 -07:00
Glenn Jocher ef75f7313d AutoAnchor n, imgsz bug fix 2020-06-16 16:48:01 -07:00
Glenn Jocher 2daa412575 train with multi-gpu half test bug fix #99 2020-06-16 16:28:25 -07:00