search

ying09 / TextFuseNet

A PyTorch implementation of "TextFuseNet: Scene Text Detection with Richer Fused Features".
MIT License
475 stars 122 forks source link

Hello, How to convert the PTH model into an ONNX format model #113

Open wangqingl opened 1 year ago

wangqingl commented 1 year ago

When I train a PTH format model and hope to convert it into an ONNX model, using ONNXRuntime for inference and deployment, I always report an error.

wangqingl commented 1 year ago

When the code runs to image = [x["iamge"].to(self.device) for x in batch_inputs] Code error “too many indices for tensor of dimension 3”