Commit Graph

147 Commits (542833c99759b727899182a2bfe2ae7a80be2d5a)

Author SHA1 Message Date
Alex Stoken 945307beba Add save_dir to plot_lr_scheduler and plot_labels
Set save_dir = log_dir in train.py
2020-06-17 16:03:18 -05:00
Alex Stoken 3263a204ea Fix get_latest_run to search 'search_dir' recursivly 2020-06-17 10:34:37 -05:00
Alex Stoken d34291733b Fix get_latest_run() to search inside 'weights' subfolders 2020-06-17 10:26:55 -05: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 1dfc28527f AutoAnchor BPR to 4 significant figures 2020-06-16 16:19:58 -07:00
Glenn Jocher 4052603e1f AutoAnchor update - improvement check 2020-06-16 15:54:20 -07:00
Glenn Jocher bafbc65ee3 AutoAnchor bug fix 2020-06-16 14:26:02 -07:00
Alex Stoken 25e51bcec7 add util function to get most recent last.pt file
added logic in train.py __main__ to handle resuming from a run
2020-06-16 15:50:27 -05:00
Alex Stoken 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
2020-06-16 15:13:03 -05: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 ec81c7b5f2 check_anchors() bug fix #90 2020-06-16 10:14:04 -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 1c0b6236e3 update fast mode 2020-06-16 00:11:29 -07:00
Lornatang 8b8b792802 There is no need to download extra packages, official bring it with you
I submitted it once in your yolov3 project, you seem to accept it? I'm not sure. I'll submit PR again.
2020-06-16 08:28:52 +08:00
Glenn Jocher db2c3acd3a updated testing settings, rebalanced towards FP16 latency 2020-06-15 16:55:57 -07:00
Glenn Jocher bd3e389130 update google_utils.py 2020-06-15 14:39:04 -07:00
Glenn Jocher c5966abba8 glob search bug fix #77 2020-06-15 12:08:57 -07:00
Lornatang 54795a4f0a fix problem 2020-06-15 16:11:06 +08:00
Glenn Jocher 260b1729f0 FP16 inference update 2020-06-14 18:32:29 -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 099e6f5ebd --img-size stride-multiple verification 2020-06-12 22:10:46 -07:00
Glenn Jocher 3a5c5328c5 ONNX, BCEBlurWithLogitsLoss, plot_study updates 2020-06-11 14:44:09 -07:00
Trevor Sullivan ac93e9c811
Fix Google Drive URI for YOLOv5 weights 2020-06-10 14:26:10 -07:00
Glenn Jocher c14368d768 new CSP model release 2020-06-09 14:31:04 -07:00
Glenn Jocher ff02ae0869 module updates 2020-06-08 22:13:01 -07:00
Glenn Jocher 1f698af6f1 create_backbone with SourceChangeWarning update 2020-06-07 15:45:05 -07:00
Glenn Jocher 8fe299f179 model fuse 2020-06-07 14:10:30 -07:00
Glenn Jocher b810b21299 augmented inference 2020-06-05 21:14:15 -07:00
Glenn Jocher c12023f145 updates 2020-06-05 20:50:43 -07:00
Glenn Jocher 66d73e4cf3 NMS updates 2020-06-05 18:44:39 -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 0362f103ef updates 2020-06-04 10:19:06 -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 ee8988b8a2 datasets.py self.batch_shapes fix 2020-06-01 18:11:18 -07:00
Glenn Jocher f96e8f3edf updates 2020-05-31 19:51:32 -07:00
Glenn Jocher b10609fe6b updates 2020-05-31 15:07:04 -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 1e84a23f38 initial commit 2020-05-29 17:04:54 -07:00