Skip to content

After converting from .pt to .torchscript, bounding box is off #6947

@willsonyee

Description

@willsonyee

Search before asking

Question

I am playing around with YOLOv5 and exporting to different formats (.pt & .torchscript). But I found that after I exported the pre-trained models using this command:
!python "/content/drive/MyDrive/yolov5/export.py" --weights "/content/drive/MyDrive/yolov5/yolov5s.pt" --include torchscript

The .torchscript model's bounding is off, please advise me on why this is happening and how to resolve it. Thank you.

.pt model:
zidane1

.torchscript model:
zidane2

Additional

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    questionFurther information is requested

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions