Text detection network psenet deployed by libtorch and Qt. （20200517调通，速度和准确度仍有待提高，gpu还没有测试过，欢迎Pr）
- ubuntu
- pytorch 1.5
- libtorch 1.5
- torchvision 0.6
- Qt5
- opencv3
Train by WenmuZhou's PSENet.pytorch codes. https://github.com/WenmuZhou/PSENet.pytorch
resnet50 and resnet152 model on icdar 2015:
bauduyun extract code: a3wx
Export model by the follow python codes, the saved model will be used in C++.
def torch_export(model, save_path):
model.eval()
data = torch.rand(1, 3, 224, 224)
traced_script_module = torch.jit.trace(model, data)
traced_script_module.save(save_path)
print("export finish.")