1172 Commits

Author	SHA1	Message	Date
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	df224a0d8f	EMA bug fix #279	2020-07-03 11:56:14 -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	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
yxNONG	53cdaf6bf5	Update utils.py	2020-07-02 13:48:19 +08:00
Glenn Jocher	9a9333d245	add replicate() to datasets.py	2020-06-29 17:10:33 -07:00
Glenn Jocher	37e13f8846	update mosaic border	2020-06-27 13:50:15 -07:00
Glenn Jocher	93a6765806	update mosaic border	2020-06-27 13:02:01 -07:00
Alex Stoken	e18e6811dc	Merge branch 'master' into advanced_logging	2020-06-27 10:13:03 -05:00
Glenn Jocher	77fb8ee082	refactor dataloader	2020-06-26 19:30:09 -07:00
Glenn Jocher	22fb2b0c25	refactor dataloader	2020-06-26 18:56:13 -07:00
Glenn Jocher	b8557f87e3	add stride to datasets.py	2020-06-24 13:02:27 -07: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	024a42f4fd	datasets.py vid_formats updated	2020-06-20 09:53:37 -07:00
Glenn Jocher	5c2cd711df	vid_formats bug fix #146	2020-06-20 09:26:07 -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	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	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