Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity
361 Commits 1 Branch 0 Tags
17b9d7cdda2fd805d2d8fccf5b88fdffca03a56c
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher 17b9d7cdda Merge pull request #344 from Laughing-q/master 
update os.sep
 2020-07-09 21:09:33 -07:00
Glenn Jocher 59beae98e8 add if isfile() to get_hash() 2020-07-09 20:57:24 -07:00
Glenn Jocher 6649770001 --rect test bug fix 2020-07-09 20:39:11 -07:00
Glenn Jocher d40e4781f1 --rect test bug patch 2020-07-09 20:36:30 -07:00
Glenn Jocher 520f5de6f0 Label caching foundational re-write #306 2020-07-09 20:07:16 -07:00
Glenn Jocher f310ca3bff Update issue templates 2020-07-09 17:36:26 -07:00
Glenn Jocher e16e9e43e1 new nc=len(names) check 2020-07-09 17:10:43 -07:00
Glenn Jocher cb527d3af9 new nc=len(names) check 2020-07-09 17:03:12 -07:00
Glenn Jocher dd8e742ece Merge pull request #334 from lorenzomammana/feature-multiple-datasets-training 
Handle multiple datasets
 2020-07-09 16:29:21 -07:00
Glenn Jocher dd33d2ab77 Update datasets.py 2020-07-09 16:28:20 -07:00
Glenn Jocher 2110f5ee89 update log_dir to runs/exp #107 2020-07-09 16:03:10 -07:00
Glenn Jocher 603ea0bfdc update log_dir to runs/exp #107 2020-07-09 15:58:07 -07:00
Glenn Jocher 72d5b58b9a disable LR plot to suppress warning message 2020-07-09 15:16:57 -07:00
Glenn Jocher 24c5a941f0 --resume EMA fix #292 2020-07-09 15:09:06 -07:00
Glenn Jocher 2b6209a9d5 Merge pull request #338 from alexstoken/hyp_save_bugfix 
Move hyp and opt yaml save to top of train()
 2020-07-09 14:35:32 -07:00
Glenn Jocher 13a09fa68a Update datasets.py 2020-07-09 13:47:20 -07:00
Glenn Jocher 47bf1730fa Update datasets.py 2020-07-09 13:45:55 -07:00
Glenn Jocher f9bab6b127 Update datasets.py 2020-07-09 13:36:23 -07:00
Glenn Jocher bf6f41567a hyperparameter printout update 2020-07-08 17:21:00 -07:00
Glenn Jocher 0fef3f6678 Merge pull request #104 from alexstoken/advanced_logging 
Log command line options, hyperparameters, and weights per run in `runs/`
 2020-07-08 17:01:32 -07:00
Glenn Jocher dc5e18390a Merge branch 'master' into advanced_logging 2020-07-08 17:01:19 -07:00
Glenn Jocher abb024de04 Delete new_hyp.yaml 2020-07-08 16:59:06 -07:00
Glenn Jocher 6b134d93c5 Update train.py 2020-07-08 16:58:13 -07:00
Glenn Jocher 22ab1c295f Update test.py 2020-07-08 16:29:31 -07:00
Glenn Jocher c3d3e6b776 Update test.py 2020-07-08 16:27:52 -07:00
Glenn Jocher cbe39a1dbb Update utils.py 2020-07-08 16:14:32 -07:00
Glenn Jocher c654d18f00 Update utils.py 2020-07-08 16:11:22 -07:00
Glenn Jocher 16f6834486 update train.py and experimental.py 2020-07-08 14:23:34 -07:00
Glenn Jocher 1b9e28e72e Update requirements.txt 2020-07-08 10:21:30 -07:00
Glenn Jocher 31b07b3550 Merge pull request #275 from wanghaoyang0106/master 
[bug fix] potential problem if img fed to model is in rectangular shape
 2020-07-08 00:07:54 -07:00
Glenn Jocher cd6609297c Merge pull request #321 from dlawrences/master 
Update CoreML model conversion to do pixel scaling as well
 2020-07-08 00:04:46 -07:00
Glenn Jocher e8cf24b6c8 Initial model ensemble capability #318 2020-07-07 15:40:50 -07:00
Glenn Jocher 121d90b3f2 update fuse_conv_and_bn() 2020-07-06 11:46:10 -07:00
Glenn Jocher 6b95d6d2c0 .to(device) bug fix 2020-07-05 23:37:36 -07:00
Glenn Jocher a62333e808 test.py .fuse() update 2020-07-05 23:27:03 -07:00
Glenn Jocher a40f615c6f .half() bug fix 2020-07-05 23:24:53 -07:00
Glenn Jocher 04bdbe4104 fuse update 2020-07-05 23:16:50 -07:00
Glenn Jocher 5ba1de0cdc update experimental.py with Ensemble() module 2020-07-05 15:02:56 -07:00
Glenn Jocher 38f5c1ad1d pruning and sparsity initial commit 2020-07-05 13:41:21 -07:00
Glenn Jocher 997ba7b346 import yaml in yolo.py 2020-07-05 12:50:04 -07:00
Glenn Jocher 8c43a6906d Update README.md 2020-07-05 12:33:51 -07:00
Glenn Jocher 2541f77946 update detect.py 2020-07-05 11:57:48 -07:00
Glenn Jocher b7fe1d0622 Merge pull request #300 from Laughing-q/master 
fix LR bug
 2020-07-05 00:42:14 -07:00
Glenn Jocher 762b06de4b Merge remote-tracking branch 'origin/master' 2020-07-04 23:07:32 -07:00
Glenn Jocher 4e2d24602d update yolo.py 2020-07-04 23:07:26 -07:00
Glenn Jocher bb3c346916 model.yaml nc inherited from dataset.yaml 2020-07-04 17:51:54 -07:00
Glenn Jocher 1fdaa4987b update detect.py 2020-07-04 17:44:53 -07:00
Glenn Jocher 11ba5294dd update get_voc.sh 2020-07-04 17:36:53 -07:00
Glenn Jocher 659ad748c8 update get_voc.sh 2020-07-04 17:13:43 -07:00
Glenn Jocher fd0b1cc924 update voc.yaml 2020-07-04 16:50:11 -07:00
Glenn Jocher 45abf0c74f remove verbose tar unzip 2020-07-04 16:42:12 -07:00
Glenn Jocher 55e48603bf udpate dataset download instructions 2020-07-04 16:25:48 -07:00
Glenn Jocher 1bc0ef874a Merge pull request #202 from ChristopherSTAN/master 
Upload some scripts about VOC dataset
 2020-07-04 16:13:15 -07:00
Glenn Jocher 72a1746938 Update get_voc.sh 2020-07-04 16:12:52 -07:00
Glenn Jocher 7df4fcebb1 Update voc.yaml 2020-07-04 16:12:17 -07:00
Glenn Jocher c0c5fd8121 Update voc.yaml 2020-07-04 16:11:14 -07:00
Glenn Jocher 9485f0b51f Merge pull request #296 from NanoCode012/patch-1 
Fix save error for multi-gpu in ema
 2020-07-04 11:50:34 -07:00
Glenn Jocher 5e2429e618 Merge remote-tracking branch 'origin/master' 2020-07-03 20:05:55 -07:00
Glenn Jocher 7bd7b2c090 update export.py 2020-07-03 20:05:50 -07:00
Glenn Jocher ce0c58f678 update compute_loss() 2020-07-03 18:56:07 -07:00
Glenn Jocher af41083d52 EMA FP16 fix #279 2020-07-03 16:57:08 -07:00
Glenn Jocher f767023c56 offset and balance update 2020-07-03 14:07:00 -07:00
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 9705c15da1 Update train.py 2020-07-02 11:37:48 -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