Hi @krish_tyger ,
- You might need to modify the
models/yolo.pyfile in the YOLOv7 repo. Kindly refer to the guide NN Example (Yolov7_tiny) to modify the forward function within class Detect. Please take note of the part of Convert PyTorch to onnx. You may also refer to this related forum thread, YOLOv7-tiny .nb (NBG) model inference shows no bounding boxes on Amb82-mini - No, you do not need to manually adjust the pegasus import script’s outputs, it’ll be automatically handled by the toolkit.
- Yes, reparameterization script is needed as the toolkit does not handle reparameterization process.
Thank you.