59 Commits

Author	SHA1	Message	Date
Glenn Jocher	1e9cf6a6e9	add file sizes to ckpt ops	2020-07-10 00:00:19 -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	bf6f41567a	hyperparameter printout update	2020-07-08 17:21:00 -07:00
Glenn Jocher	dc5e18390a	Merge branch 'master' into advanced_logging	2020-07-08 17:01:19 -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	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
Alex Stoken	95f0a56df7	Bug fix to get_latest_run when recent run is named	2020-07-07 11:19:49 -05:00
Alex Stoken	f517ba81c7	Merge branch 'master' into advanced_logging	2020-07-06 16:52:11 -05:00
Glenn Jocher	ce0c58f678	update compute_loss()	2020-07-03 18:56:07 -07:00
Glenn Jocher	f767023c56	offset and balance update	2020-07-03 14:07:00 -07:00
wanghaoyang0106	b57f83d005	[bug fix] potential problem if img fed to model is in rectangular shape	2020-07-03 13:09:21 +08:00
Glenn Jocher	4e62eb90be	check_img_size() int() update	2020-06-30 16:41:26 -07:00
Alex Stoken	e18e6811dc	Merge branch 'master' into advanced_logging	2020-06-27 10:13:03 -05:00
Glenn Jocher	3b062254a6	update utils.create_pretrained()	2020-06-24 12:53:43 -07:00
Glenn Jocher	6c1b87a42e	update google_utils import	2020-06-22 23:00:23 -07:00
Glenn Jocher	364fcfd7db	PANet update	2020-06-22 14:27:17 -07:00
Glenn Jocher	fc171e2678	check_anchor_order() update	2020-06-21 14:08:59 -07:00
Glenn Jocher	1f1917ef56	remove fast, add merge	2020-06-21 13:37:11 -07:00
Alex Stoken	e572bb0803	Add plot_results save location to log_dir	2020-06-21 09:36:28 -05: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	9fdb0fbacf	AutoAnchor bug fix # 117	2020-06-17 19:51:15 -07: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	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	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
Glenn Jocher	db2c3acd3a	updated testing settings, rebalanced towards FP16 latency	2020-06-15 16:55:57 -07:00
Glenn Jocher	c5966abba8	glob search bug fix #77	2020-06-15 12:08:57 -07: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
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