MegEngine / YOLOX

MegEngine implementation of YOLOX
Apache License 2.0
106 stars 15 forks source link

Something wrong in yolo_head.py #9

Closed AlexZen1204 closed 2 years ago

AlexZen1204 commented 2 years ago

image it should be like this : yv, xv = meshgrid...

FateScript commented 2 years ago

Could you also tell me the reason of doing this?

AlexZen1204 commented 2 years ago

Could you also tell me the reason of doing this? “def get_in_boxes_info” gets x_shifts and y_shifts from “def get_output_and_grid”. If the input has the same length and width, it's ok, or it gets wrong shifts. More, this is the “def get_output_and_grid” in yolox of torch version . image