Commit Graph
97 Commits
Author SHA1 Message Date
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
Glenn Jocher 546f973562 model fuse 2020-06-07 14:47:48 -07:00
Glenn Jocher 8fe299f179 model fuse 2020-06-07 14:10:30 -07:00
Glenn Jocher c672bef10f model fuse 2020-06-07 14:07:22 -07:00
Glenn Jocher 12b0c046d5 model fusion and onnx export 2020-06-07 13:42:33 -07:00
Glenn Jocher b810b21299 augmented inference 2020-06-05 21:14:15 -07:00
Glenn Jocher d5d160449c updates 2020-06-05 21:03:17 -07:00
Glenn Jocher c12023f145 updates 2020-06-05 20:50:43 -07:00
Glenn Jocher 5c470d235e multi-gpu test bug fix #15 2020-06-05 20:36:29 -07:00
Glenn Jocher 67d5e65c77 profiling update 2020-06-05 20:22:39 -07:00
Glenn Jocher 2703ac73e8 profiling update 2020-06-05 20:22:17 -07:00
Glenn Jocher 4e32b6029a update 2020-06-05 20:03:21 -07:00
Glenn Jocher 66d73e4cf3 NMS updates 2020-06-05 18:44:39 -07:00
Glenn Jocher 208493d959 updates 2020-06-05 16:34:05 -07:00
Glenn Jocher 948bcdd219 --classes bug fix #17 2020-06-05 13:29:26 -07:00
Glenn Jocher 22d6088205 speed-reproducibility fix #17 2020-06-05 13:07:09 -07:00
Glenn Jocher 55ca5c74d2 multi-scale fix #16 2020-06-05 12:57:16 -07:00
Glenn Jocher 7c2832cd49 assert equal model and dataset classes 2020-06-04 17:21:22 -07:00
Glenn Jocher 11121e39ed updates 2020-06-04 16:52:07 -07:00
Glenn Jocher dbdee3a4a3 multi-gpu test bug fix #15 2020-06-04 15:23:03 -07:00
Glenn Jocher a2336088f0 updates 2020-06-04 14:25:10 -07:00
Glenn Jocher ba9ab663a2 model.names for dataParallel 2020-06-04 13:30:14 -07:00
Glenn Jocher 0362f103ef updates 2020-06-04 10:19:06 -07:00
Glenn Jocher 94e67115db updates 2020-06-03 14:48:32 -07:00
Glenn Jocher 2a0aff685a dataset padding mode 2020-06-03 13:55:33 -07:00
Glenn Jocher eb97b2e413 NMS fast mode 2020-06-03 13:02:59 -07:00
Glenn Jocher d97d31ed6b updates 2020-06-03 09:27:43 -07:00
Glenn Jocher 391492ee5b updates 2020-06-02 23:57:58 -07:00
Glenn Jocher 17e4926cdf updates 2020-06-02 23:40:14 -07:00
Glenn Jocher 11b462e2a9 remove thop dependency 2020-06-02 11:29:39 -07:00
Glenn Jocher ee8988b8a2 datasets.py self.batch_shapes fix 2020-06-01 18:11:18 -07:00
Glenn Jocher 7a00a5eff2 onnx_export.py 2020-06-01 14:05:01 -07:00
Glenn Jocher df7988d8d0 onnx_export.py 2020-06-01 13:53:47 -07:00
Glenn Jocher 8cab44e78b Merge remote-tracking branch 'origin/master' 2020-05-31 19:51:39 -07:00
Glenn Jocher f96e8f3edf updates 2020-05-31 19:51:32 -07:00
Glenn Jocher f788a6af96 Update README.md 2020-05-31 16:50:48 -07:00
Glenn Jocher 76d879e8d0 Update README.md 2020-05-31 16:45:12 -07:00
Glenn Jocher b10609fe6b updates 2020-05-31 15:07:04 -07:00
Glenn Jocher 7a61d524d3 updates 2020-05-31 11:06:13 -07:00
Glenn Jocher cf3ee51b8d updates 2020-05-31 11:02:40 -07:00
Glenn Jocher f688b45712 updates 2020-05-31 11:00:09 -07:00
Glenn Jocher 1500775463 Update greetings.yml 2020-05-31 09:30:50 -07:00
Glenn Jocher 72eed215ff updates 2020-05-31 09:25:42 -07:00
Glenn Jocher 5a79b5f65c updates 2020-05-30 15:22:09 -07:00
Glenn Jocher 4f44aaf26b updates 2020-05-30 15:18:59 -07:00
Glenn Jocher f129cdda9f updates 2020-05-30 11:03:50 -07:00
Glenn Jocher 1b97392801 updates 2020-05-30 10:46:35 -07:00
Glenn Jocher a88cdd25d7 updates 2020-05-30 10:34:43 -07:00
Glenn Jocher 4f00201a44 updates 2020-05-30 09:58:02 -07:00
Glenn Jocher 9bb5968ae5 updates 2020-05-30 09:56:15 -07:00
Glenn Jocher 893a9059e9 updates 2020-05-30 09:49:11 -07:00
Glenn Jocher 7960c4c5d8 updates 2020-05-30 09:28:11 -07:00
Glenn Jocher ce36905358 updates 2020-05-30 00:12:45 -07:00
Glenn Jocher 1e84a23f38 initial commit 2020-05-29 17:04:54 -07:00