SkalskiP/YOLO-World · adaptive_max_pool2d not supported by ONNX exporter and question about classes.

Thanks for your great work, it allow me to understand better the structure.

I tried to export to onnx based in your code, the biggest issue is that the models used by your example are using adaptive_max_pool2d and that is not supported by ONNX. I don't see that op in the original .pth, is this added by you in the final .pt model file at exporting or is original from the .pth ? there is anyway to export a model that doesn't use it? Why the pt model is 1/4 size of the original pth?

another thing, i don't understand where and how the classes are set as inputs to the mode. i saw the set_classes function and trace it, but as far i understand, the model only get's an input of the image as 1,3,640,640 right?

Thanks!