You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
pangchao-git
changed the title
mmdetection masklrcnn to deepstream, 如何支持在deepstream获取mask
mmdetection maskrcnn to deepstream, 如何支持在deepstream获取mask
May 30, 2022
1 模型转化时候这样修改了Infrence.py
trt_model = mmdet2trt(cfg_path, args.checkpoint, output_names=["num_detections", "boxes", "scores", "classes","masks"], fp16_mode=args.fp16, device=args.device,enable_mask=True)
torch.save(trt_model.state_dict(), args.save_path)
2 deepstream 的配置文件
output-bbox-name=bbox
output-blob-names=num_detections;boxes;scores;classes;masks
parse-bbox-func-name=NvDsInferParseMmdet
custom-lib-path=nvdsinfer_custom_impl_fasterRCNN/libamirstan_plugin.so
The text was updated successfully, but these errors were encountered: