31 Commits

Author	SHA1	Message	Date
Alex Stoken	d64ad0fbf3	Remove --resume functionality and related checks/logic.	2020-06-24 09:17:27 -05:00
Alex Stoken	e572bb0803	Add plot_results save location to log_dir	2020-06-21 09:36:28 -05: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
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	5f2eeba233	remove old print statements	2020-06-16 17:09:39 -05: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
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
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	8b26e89006	AutoAnchor bug fix #72	2020-06-16 00:53:34 -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	915b1481fc	default check_git_status() to True	2020-06-15 16:18:46 -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	31f3310029	assert best possible recall > 0.9 before training	2020-06-13 15:05:41 -07:00
Lauritzen Kasper Primdal	c3d4d321d3	Ensures weights/ dir exists ... Allows train.py to be run outside of yolov5/ directory.	2020-06-13 14:44:23 +02:00
Glenn Jocher	099e6f5ebd	--img-size stride-multiple verification	2020-06-12 22:10:46 -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	eb97b2e413	NMS fast mode	2020-06-03 13:02:59 -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