6 Commits (dbf160423135c9f69a4923567ac6e5963812b282)

Author	SHA1	Message	Date
liaoxingyu	dbf1604231	fix model deploy problems	2021-05-25 15:55:43 +08:00
Xingyu Liao	be0a089e1f	bugfix & merge classification transforms (#448) ... Summary: change heads definition in project and config file, merge classification transforms into default transforms	2021-03-30 15:47:14 +08:00
liaoxingyu	9d83550b67	bugfix ... Summary: make `deploy` adaptation for other models except repvgg	2021-03-26 10:10:45 +08:00
Xingyu Liao	9b5af4166e	support repvgg (#429) ... Summary: * support repvgg backbone, and verify the consistency of train mode and eval mode * onnx export logger style modification	2021-03-11 14:23:24 +08:00
Xingyu Liao	cb7a1cb3e1	update deployment toolchain (#428) ... Summary: Remove tiny-tensorrt dependency and rewrite a new tensorrt inference api. In the new version of trt infer, it can pad the input to fixed batch automatically, so you don't need to worry about dynamic batch size.	2021-03-10 16:48:59 +08:00
liaoxingyu	16655448c2	onnx/trt support ... Summary: change model pretrain mode and support onnx/TensorRT export	2020-07-29 17:43:39 +08:00