2799 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
ashing	62eda5e9c2	add "__init__.py" file in <models> folder	2020-06-15 14:37:07 +08:00
Glenn Jocher	111994956e	switch default inference to FP16 on GPU	2020-06-14 19:25:39 -07:00
Glenn Jocher	157aff2854	FP16 inference fix	2020-06-14 18:45:20 -07:00
Glenn Jocher	c6fc4242a2	Merge remote-tracking branch 'origin/master'	2020-06-14 18:32:35 -07:00
Glenn Jocher	260b1729f0	FP16 inference update	2020-06-14 18:32:29 -07:00
Glenn Jocher	b545ee32f4	Created using Colaboratory	2020-06-14 13:29:05 -07:00
Glenn Jocher	629d3701fc	pyyaml >= 5.3	2020-06-14 12:42:51 -07:00
Glenn Jocher	4b0e3745f2	Merge pull request #59 from Lijun-Yu/master ... Fix config path for PyTorch Hub	2020-06-14 12:16:41 -07:00
Glenn Jocher	4f6c0cf4c5	Update hubconf.py	2020-06-14 12:15:14 -07:00
Lijun Yu	0dec8ffd66	path fix for torch hub	2020-06-14 14:47:40 -04:00
Glenn Jocher	4dd75b3635	pycocotools git install fix	2020-06-14 10:54:00 -07:00
Glenn Jocher	9bf35e7b86	Merge pull request #56 from channeladam/master ... Fix initial basic setup / new starter issues	2020-06-14 10:33:34 -07:00
Glenn Jocher	335c607a24	Update datasets.py	2020-06-14 10:32:59 -07:00
Glenn Jocher	1374581929	Update requirements.txt	2020-06-14 10:23:33 -07:00
Adam	f47ca2085c	Include yolov5x extra large model weights	2020-06-14 18:19:23 +10:00
Adam	34bf936ce6	Add more logging for diagnostics	2020-06-14 18:18:57 +10:00
Adam	faa6aeaa20	Fix pip requirements issues	2020-06-14 18:18:31 +10: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	19e68e8a7b	Merge pull request #51 from KPLauritzen/b-ensure-weights-dir-when-training ... Ensures weights/ dir exists	2020-06-13 12:54:43 -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	4ba47b1b1b	model.yaml updates	2020-06-12 15:41:46 -07:00
Glenn Jocher	1e2cb6b480	ONNX export explicit cpu map_location	2020-06-12 13:54:00 -07:00
Glenn Jocher	ad71d2d513	update times to end-to-end latency (inference + NMS)	2020-06-12 11:42:08 -07:00
Glenn Jocher	8f17a96ff0	cleanup	2020-06-11 21:55:22 -07:00
Glenn Jocher	8f1755290c	ONNX export update	2020-06-11 21:54:01 -07:00
Glenn Jocher	3a5c5328c5	ONNX, BCEBlurWithLogitsLoss, plot_study updates	2020-06-11 14:44:09 -07:00
Glenn Jocher	18b4da91e0	PyTorch Hub update	2020-06-10 19:14:47 -07:00
Glenn Jocher	c8f0694894	PyTorch Hub update	2020-06-10 19:13:04 -07:00
Glenn Jocher	a814720403	PyTorch Hub updates	2020-06-10 19:11:11 -07:00
Glenn Jocher	1e25775c7d	Merge remote-tracking branch 'origin/master'	2020-06-10 17:09:43 -07:00
Glenn Jocher	0c7a0a08e9	update greetings.yml	2020-06-10 17:09:37 -07:00
Glenn Jocher	f3daa50a32	Update README.md	2020-06-10 17:07:25 -07:00
Glenn Jocher	46e3cadf28	Update README.md	2020-06-10 16:22:18 -07:00
Glenn Jocher	a4ae9a6850	Update README.md	2020-06-10 15:56:03 -07:00
Glenn Jocher	bb56ed4d05	Merge pull request #35 from pcgeek86/patch-1 ... Fix Google Drive URI for YOLOv5 weights	2020-06-10 14:42:50 -07:00
Trevor Sullivan	ac93e9c811	Fix Google Drive URI for YOLOv5 weights	2020-06-10 14:26:10 -07:00
Glenn Jocher	9b62f9962e	bug fix #30	2020-06-10 00:44:34 -07:00
Glenn Jocher	c14368d768	new CSP model release	2020-06-09 14:31:04 -07:00
Glenn Jocher	854ccb9a74	_print_weights()	2020-06-09 10:51:47 -07:00
Glenn Jocher	5bee686649	model definition update	2020-06-08 23:26:56 -07:00
Glenn Jocher	715cb08b10	module updates	2020-06-08 22:15:16 -07:00
Glenn Jocher	ff02ae0869	module updates	2020-06-08 22:13:01 -07:00
Glenn Jocher	d187459dd4	study changes	2020-06-08 10:42:33 -07:00
Glenn Jocher	1f698af6f1	create_backbone with SourceChangeWarning update	2020-06-07 15:45:05 -07:00
Glenn Jocher	546f973562	model fuse	2020-06-07 14:47:48 -07:00
Glenn Jocher	8fe299f179	model fuse	2020-06-07 14:10:30 -07:00
Glenn Jocher	c672bef10f	model fuse	2020-06-07 14:07:22 -07:00
Glenn Jocher	12b0c046d5	model fusion and onnx export	2020-06-07 13:42:33 -07:00