Post processing in C++ tensorrt #1

Open

mahmoodkashmiri

opened

on Mar 25, 2024

Hi, I know already a great piece of work thanks for that. Could you please guide how to post process the output segmentation image in C++ cuda api (tensorrt)? in python we use torch.argmax() and interpolation for post processing.

to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests