mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
570 Commits 10 Branches 10 Tags
Commit Graph

570 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
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
Alex Stoken
c8152c81a6 Syntax fixes 2020-06-17 16:32:13 -05:00
Alex Stoken
9b7386f603 Add save_dir arg to test.test, use arg as location for saving batch jpgs 2020-06-17 16:08:46 -05:00
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
3b2b330872 Move results.txt from weights/ to log_dir 2020-06-17 15:55:45 -05: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
Alex Stoken
ade023cff2 Fix hyp file read in and dict update. 
Add example of hyp yaml
 2020-06-17 10:59:20 -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
91de3381e1
Merge pull request #112 from imprashr/patch-1 
onnx should be 1.6.0
 2020-06-17 08:23:35 -07:00
Prashant Rai
0071b69994
onnx should be 1.6.0 
for smoothly export model to onnx, onnx should be 1.6.0
 2020-06-17 14:15:27 +05:30
Lornatang
2368603484 fix refrence bug 
In torch==1.5, the import of the API has changed. Although it does not interrupt the operation of the program, it seems to me to be an implicit error and may throw an exception in later versions.
 2020-06-17 09:56:26 +08: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
Alex Stoken
5f2eeba233 remove old print statements 2020-06-16 17:09:39 -05: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
Alex Stoken
333f678b37 add update default hyp dict with provided yaml 2020-06-16 16:36:20 -05:00
Alex Stoken
a448c3bcd7 add logic for resuming and getting hyp for resume run 2020-06-16 16:30:12 -05: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
Alex Stoken
4418809cf5 change weights dir (wdir) to be unique to each run, under log_dir 2020-06-16 15:09:51 -05:00
Alex Stoken
d9f446cd81 add save yaml of opt and hyp to tensorboard log_dir in train() 2020-06-16 15:06:13 -05:00
Alex Stoken
a85e6d0fc0 add parser arg for hyp yaml file 2020-06-16 14:53:32 -05: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