982 Commits

Author	SHA1	Message	Date
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
Glenn Jocher	e8cf24b6c8	Initial model ensemble capability #318	2020-07-07 15:40:50 -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	121d90b3f2	update fuse_conv_and_bn()	2020-07-06 11:46:10 -07:00
Glenn Jocher	38f5c1ad1d	pruning and sparsity initial commit	2020-07-05 13:41:21 -07:00
NanoCode012	ccfa9373ae	Update to remove "module" attribute check	2020-07-04 23:12:57 +07:00
NanoCode012	f83521ab9a	Fix save error for multi-gpu	2020-07-04 16:50:33 +07:00
Glenn Jocher	ce0c58f678	update compute_loss()	2020-07-03 18:56:07 -07:00
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