I am using a C++ program to analyze a CNN (YoloV5x). My goal is to extract built-in function parameters, like stride length and kernel size from the model file. I am currently using the
torch::jit::script::Module data type and
torch::jit::load function to load the torchscript model.
My problem is that I see no way of accessing attributes, even though software like “Netron” can obviously extract that. The only remotely related feature is
model.named_parameters(), which only gives weights and biases of the model, which is not enough for my analysis. Is there any programmatic way to extract the needed features of the built-in functions using the C++ API?
When I load the model (model.pt) in python and print it, one of the function ouptuts is the following:
(1): Conv( (conv): Conv2d(80, 160, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1)) (act): Hardswish() )
I would like to get the
padding parameter in C++ API.
In the Python program I load model.pt file and in the C++ file I load the model.torchscript.pt file.