Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix HRNet dimension error on images with alpha channel (#5570)
This is pretty much the same fix applied to f-BRS in #5384. We've been using HRNet for a while and now an then we receive "500 errors" just as reported in #5299 when someone forgets to drop the alpha-channel from our images. ### Motivation and context The RuntimeError is a little bit different, but comes from the same issue: RGBA instead of RGB images: > RuntimeError: Given groups=1, weight of size [16, 3, 1, 1], expected input[*, 4, *, *] to have 3 channels, but got 4 channels instead. ### How has this been tested? I created a task with images with and w/o alpha channel, and the interactor works on both now.
- Loading branch information