Open Strike1999 opened 2 weeks ago
Thanks for your work!
For image segmentation, you take
image = normalize(image, (0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
. However, when inferencing reward modeloutputs = reward_model(image.to(accelerator.device))
, does this step inherently include image normalization as part of the MMsegmentation test pipeline?Thanks again.
Thank you for your attention. We call the seg model forward function that defined in MMSeg, instead of directly calling the entire MMSeg Pipeline for inference, so it will not be affected by normalization.
Thanks for your work!
For image segmentation, you take
image = normalize(image, (0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
. However, when inferencing reward modeloutputs = reward_model(image.to(accelerator.device))
, does this step inherently include image normalization as part of the MMsegmentation test pipeline?Thanks again.