SOLOv2.tensorRT
SOLOv2.tensorRT copied to clipboard
zhangjinsong3
SOLOv2 on onnx & tensorRT
I run `python deploy/onnx_exporter_solov2.py configs/solov2/solov2_light_448_r18_fpn_8gpu_3x.py weights/SOLOv2_LIGHT_448_R18_3x.onnx --checkpoint ./checkpoints/SOLOv2_LIGHT_448_R18_3x.pth --shape 448 672` and got, `Traceback (most recent call last): File "deploy/onnx_exporter_solov2.py", line 225, in convert2onnx(args, dummy_input) File "deploy/onnx_exporter_solov2.py", line 100, in...
I try to setup this repository for the conversion of a trained model. I do however not succeed in getting an environment working with all required dependency versions. Is there...
What's the advantage of TensorRT compared with pytorch?Both 18ms. Does this mean that TensorRT has very limited optimization for SOLOV2?
你好，我想问一下这个output_names为什么由11个改成3个了，如果还是原来的11个，该怎么修改inference_on_onnxrt_trt_solov2.py 谢谢
I have been training a SOLOv2 and a Solov2 light model on the Resnet 50 backbone. I was wondering if this codebase can be used to convert the model to...
Traceback (most recent call last): File "onnx_exporter.py", line 232, in check(args, dummy_input, check_onnx=True, check_trt=False) File "onnx_exporter.py", line 127, in check sess = rt.InferenceSession(args.out) File "/home/tao/anaconda3/lib/python3.8/site-packages/onnxruntime/capi/session.py", line 195, in __init__ self._create_inference_session(providers,...
你好，请教你在转onnx模型时，将solov2_head.py中get_seg() 如下操作放入onnx model，这个对应要改哪个地方？
I want to use " multiple images per gpu".in "# TODO: support multiple images per gpu (only minor changes are needed) dataset = build_dataset(cfg.data.test) data_loader = build_dataloader( dataset, imgs_per_gpu=2, workers_per_gpu=1,...
