Commit Graph

32 Commits (02445d176d3a8756fd5f2800b66137beeac9e573)

Author SHA1 Message Date
Glenn Jocher 02445d176d improved model.yaml source tracking 2020-07-13 14:35:47 -07:00
Glenn Jocher 78c9420e09 --classes bug fix #387 2020-07-13 13:27:46 -07:00
Glenn Jocher 094079bd26 update detect.py --class argument #387 2020-07-13 10:53:34 -07:00
Glenn Jocher e8cf24b6c8 Initial model ensemble capability #318 2020-07-07 15:40:50 -07:00
Glenn Jocher a40f615c6f .half() bug fix 2020-07-05 23:24:53 -07:00
Glenn Jocher 04bdbe4104 fuse update 2020-07-05 23:16:50 -07:00
Glenn Jocher 2541f77946 update detect.py 2020-07-05 11:57:48 -07:00
Glenn Jocher 1fdaa4987b update detect.py 2020-07-04 17:44:53 -07:00
Glenn Jocher 659ad748c8 update get_voc.sh 2020-07-04 17:13:43 -07:00
Glenn Jocher 5323ad224d export.py update 2020-07-01 16:15:25 -07:00
Glenn Jocher a62a1c2c67 export.py update 2020-07-01 16:14:49 -07:00
Glenn Jocher a9dc0c2c29 update check_img_size() for model strides 2020-06-30 15:25:38 -07:00
Russ Ferriday 0d5065d3b9
Remove non-ascii that breaks pdb single-stepping 2020-06-29 12:56:40 +01:00
Glenn Jocher 8c5b6220f5
Update detect.py 2020-06-25 19:26:23 -07:00
Glenn Jocher 496ec33a33
Update detect.py
Added some recent updates that were missing, and updated the filename with an if else.
2020-06-25 19:19:15 -07:00
Anish Hiranandani 68f63616b3 Fix after suggestion 2020-06-25 18:23:36 -07:00
Glenn Jocher 6c1b87a42e update google_utils import 2020-06-22 23:00:23 -07:00
Glenn Jocher bbd12c7cfe detect.py fast bug fix 2020-06-22 10:14:58 -07:00
Lornatang 2368603484 fix refrence bug
In torch==1.5, the import of the API has changed. Although it does not interrupt the operation of the program, it seems to me to be an implicit error and may throw an exception in later versions.
2020-06-17 09:56:26 +08:00
Glenn Jocher 76ca367a01 FP16 to FP32 ckpt load 2020-06-15 13:32:45 -07:00
Glenn Jocher 111994956e switch default inference to FP16 on GPU 2020-06-14 19:25:39 -07:00
Glenn Jocher 260b1729f0 FP16 inference update 2020-06-14 18:32:29 -07:00
Glenn Jocher 099e6f5ebd --img-size stride-multiple verification 2020-06-12 22:10:46 -07:00
Glenn Jocher ad71d2d513 update times to end-to-end latency (inference + NMS) 2020-06-12 11:42:08 -07:00
Glenn Jocher c14368d768 new CSP model release 2020-06-09 14:31:04 -07:00
Glenn Jocher c672bef10f model fuse 2020-06-07 14:07:22 -07:00
Glenn Jocher dbdee3a4a3 multi-gpu test bug fix #15 2020-06-04 15:23:03 -07:00
Glenn Jocher ba9ab663a2 model.names for dataParallel 2020-06-04 13:30:14 -07:00
Glenn Jocher eb97b2e413 NMS fast mode 2020-06-03 13:02:59 -07:00
Glenn Jocher 391492ee5b updates 2020-06-02 23:57:58 -07:00
Glenn Jocher df7988d8d0 onnx_export.py 2020-06-01 13:53:47 -07:00
Glenn Jocher 1e84a23f38 initial commit 2020-05-29 17:04:54 -07:00