Referring to the above article, I know that the onnx model file used in sample 04 of MMAPI is resnet10 object detector.
File name: resnet10_dynamic_batch.onnx
In what framework was this model trained before conversion to onnx? (It seems to have been learned in TLT as expected.)
- If the learning framework is Pytorch, you need to use the torch.onnx.export function to convert it to onnx.
model = network class
I know I need to create a model object from it.
I would like to know the network definition for creating this model object.
- In case of non-Patorch, can you tell me how to learn the model as a custom dataset and the code to convert it to onnx?