mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
Corrected the tensor shape in the doc string. The incoming masks are stacked in dim=0 therefore the doc is wrong Signed-off-by: Jonas Heinle <jonasheinle@googlemail.com>