Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
167 Commits 1 Branch 0 Tags
364fcfd7dba53f46edd4f04c037a039c0a287972
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
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
Glenn Jocher d1e5716610 train with multi-gpu half test bug fix #99 2020-06-16 16:22:58 -07:00
Glenn Jocher 1dfc28527f AutoAnchor BPR to 4 significant figures 2020-06-16 16:19:58 -07:00
Glenn Jocher 8db51c7002 tb_writer bug fix 2020-06-16 16:05:28 -07:00
Glenn Jocher 4052603e1f AutoAnchor update - improvement check 2020-06-16 15:54:20 -07:00
Glenn Jocher afe1df385b dist.destroy_process_group() bug fix 2020-06-16 15:08:14 -07:00
Glenn Jocher 72d06147b2 Multi-GPU disabled in test.py 2020-06-16 14:42:30 -07:00
Glenn Jocher bafbc65ee3 AutoAnchor bug fix 2020-06-16 14:26:02 -07:00
Glenn Jocher 57a0ae3350 AutoAnchor implementation 2020-06-16 13:12:22 -07:00
Glenn Jocher 95c46f7245 Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library 
Delete redundant thrid party library
 2020-06-16 11:52:36 -07:00
Glenn Jocher 5a50491fa1 check_anchors bug fix 2020-06-16 10:36:35 -07:00
Glenn Jocher 05b8ee5ca4 check_anchors() bug fix #102 2020-06-16 10:34:16 -07:00
Glenn Jocher 7c6fd4d670 --verbose bug fix #84 2020-06-16 10:22:22 -07:00
Glenn Jocher ec81c7b5f2 check_anchors() bug fix #90 2020-06-16 10:14:04 -07:00
Glenn Jocher e1e3399286 ONNX export bug fix #93 2020-06-16 09:59:42 -07:00
Glenn Jocher 8b26e89006 AutoAnchor bug fix #72 2020-06-16 00:53:34 -07:00
Glenn Jocher 8fa3724072 kmean_anchors() update 2020-06-16 00:40:21 -07:00
Glenn Jocher bdd9fee841 update fast mode 2020-06-16 00:29:54 -07:00
Glenn Jocher 1c0b6236e3 update fast mode 2020-06-16 00:11:29 -07:00
Glenn Jocher db2c3acd3a updated testing settings, rebalanced towards FP16 latency 2020-06-15 16:55:57 -07:00
Glenn Jocher 915b1481fc default check_git_status() to True 2020-06-15 16:18:46 -07:00
Glenn Jocher 0b514daced FP16 test loss bug fix 2020-06-15 15:26:29 -07:00
Glenn Jocher a1748a8d6e test during training default to FP16 2020-06-15 15:20:27 -07:00
Glenn Jocher bd3e389130 update google_utils.py 2020-06-15 14:39:04 -07:00
Glenn Jocher 76ca367a01 FP16 to FP32 ckpt load 2020-06-15 13:32:45 -07:00
Glenn Jocher 6fb5ff014c FP16 to FP32 ckpt load 2020-06-15 13:21:34 -07:00
Glenn Jocher 14523bb030 FP16 to FP32 ckpt load 2020-06-15 13:18:39 -07:00
Glenn Jocher c5966abba8 glob search bug fix #77 2020-06-15 12:08:57 -07:00
Glenn Jocher ace56eb5b6 Merge pull request #70 from Lornatang/fix-reference-bugs 
fix reference problem
 2020-06-15 11:18:17 -07:00
Glenn Jocher 45a17ea40c Merge remote-tracking branch 'origin/master' 2020-06-15 00:44:08 -07:00
Glenn Jocher e7c24e04ac half() fix #66 2020-06-15 00:43:57 -07:00
Glenn Jocher 6570336284 Merge pull request #65 from Ashing00/master 
add "__init__.py" file in <models> folder
 2020-06-15 00:34:54 -07:00
Glenn Jocher 111994956e switch default inference to FP16 on GPU 2020-06-14 19:25:39 -07:00
Glenn Jocher 157aff2854 FP16 inference fix 2020-06-14 18:45:20 -07:00
Glenn Jocher c6fc4242a2 Merge remote-tracking branch 'origin/master' 2020-06-14 18:32:35 -07:00
Glenn Jocher 260b1729f0 FP16 inference update 2020-06-14 18:32:29 -07:00
Glenn Jocher b545ee32f4 Created using Colaboratory 2020-06-14 13:29:05 -07:00
Glenn Jocher 629d3701fc pyyaml >= 5.3 2020-06-14 12:42:51 -07:00
Glenn Jocher 4b0e3745f2 Merge pull request #59 from Lijun-Yu/master 
Fix config path for PyTorch Hub
 2020-06-14 12:16:41 -07:00
Glenn Jocher 4f6c0cf4c5 Update hubconf.py 2020-06-14 12:15:14 -07:00
Glenn Jocher 4dd75b3635 pycocotools git install fix 2020-06-14 10:54:00 -07:00
Glenn Jocher 9bf35e7b86 Merge pull request #56 from channeladam/master 
Fix initial basic setup / new starter issues
 2020-06-14 10:33:34 -07:00
Glenn Jocher 335c607a24 Update datasets.py 2020-06-14 10:32:59 -07:00
Glenn Jocher 1374581929 Update requirements.txt 2020-06-14 10:23:33 -07:00
Glenn Jocher 93bbf1b87a best possible recall fix 2020-06-13 15:25:53 -07:00
Glenn Jocher 31f3310029 assert best possible recall > 0.9 before training 2020-06-13 15:05:41 -07:00
Glenn Jocher 19e68e8a7b Merge pull request #51 from KPLauritzen/b-ensure-weights-dir-when-training 
Ensures weights/ dir exists
 2020-06-13 12:54:43 -07:00
Glenn Jocher 099e6f5ebd --img-size stride-multiple verification 2020-06-12 22:10:46 -07:00
Glenn Jocher 4ba47b1b1b model.yaml updates 2020-06-12 15:41:46 -07:00
Glenn Jocher 1e2cb6b480 ONNX export explicit cpu map_location 2020-06-12 13:54:00 -07:00
Glenn Jocher ad71d2d513 update times to end-to-end latency (inference + NMS) 2020-06-12 11:42:08 -07:00
Glenn Jocher 8f17a96ff0 cleanup 2020-06-11 21:55:22 -07:00
Glenn Jocher 8f1755290c ONNX export update 2020-06-11 21:54:01 -07:00
Glenn Jocher 3a5c5328c5 ONNX, BCEBlurWithLogitsLoss, plot_study updates 2020-06-11 14:44:09 -07:00
Glenn Jocher 18b4da91e0 PyTorch Hub update 2020-06-10 19:14:47 -07:00
Glenn Jocher c8f0694894 PyTorch Hub update 2020-06-10 19:13:04 -07:00
Glenn Jocher a814720403 PyTorch Hub updates 2020-06-10 19:11:11 -07:00
Glenn Jocher 1e25775c7d Merge remote-tracking branch 'origin/master' 2020-06-10 17:09:43 -07:00
Glenn Jocher 0c7a0a08e9 update greetings.yml 2020-06-10 17:09:37 -07:00
Glenn Jocher f3daa50a32 Update README.md 2020-06-10 17:07:25 -07:00
Glenn Jocher 46e3cadf28 Update README.md 2020-06-10 16:22:18 -07:00
Glenn Jocher a4ae9a6850 Update README.md 2020-06-10 15:56:03 -07:00
Glenn Jocher bb56ed4d05 Merge pull request #35 from pcgeek86/patch-1 
Fix Google Drive URI for YOLOv5 weights
 2020-06-10 14:42:50 -07:00
Glenn Jocher 9b62f9962e bug fix #30 2020-06-10 00:44:34 -07:00
Glenn Jocher c14368d768 new CSP model release 2020-06-09 14:31:04 -07:00
Glenn Jocher 854ccb9a74 _print_weights() 2020-06-09 10:51:47 -07:00
Glenn Jocher 5bee686649 model definition update 2020-06-08 23:26:56 -07:00
Glenn Jocher 715cb08b10 module updates 2020-06-08 22:15:16 -07:00
Glenn Jocher ff02ae0869 module updates 2020-06-08 22:13:01 -07:00
Glenn Jocher d187459dd4 study changes 2020-06-08 10:42:33 -07:00
Glenn Jocher 1f698af6f1 create_backbone with SourceChangeWarning update 2020-06-07 15:45:05 -07:00