Peng Chen
@stan-haochen If you can provide pretrained mode with the BN normalization in the head, I could pay some effort to convert Blendmask into other libraries or SDK such as ONNX/ncnn/caffe.
@jinfagang Sorry for the late rely I'm very willing to do the convert if you could provide the BN head model.
@jinfagang @snaillp I've discussed with @stan-haochen. We would pay some effort on training and converting the model.
@jinfagang Thanks for the training file. I'm working on the covert.
@all I almost finished the convert to onnx. Verfication code in the onnxruntime is also supported. Here 'almost' indicates most of the result are correct tested in onnxruntime for the...
yes, the both verification in tensorrt and onnxruntime are added. check the new PR, please
@jinfagang I found I made a mistake by take the caffe2 engine, I'm swtiching to the TensorRT engine
@jinfagang I already block those unsupport layers when exporting the onnx file, such as RoiAlign and NMS. I fix the onnx verification script and add an one-in-all script demo `pytorch-onnx-caffe-ncnn-rt.sh`...
Yes. As those layers seem not to be standard supported in many frameworks and even some frameworks have the support but with a different implementation, layers such as ROIAlign and...
@jinfagang I only test the onnx file by the onnx_tensorrt package. Refer line 270/271 in https://github.com/blueardour/uofa-AdelaiDet/blob/master/onnx/test_onnxruntime.py I didn't test in a standalone tensorrt. Ps. My PR seems not approved yet....