You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The CocoDetection v2 transform wrapper attempts to decode the mask itself, but pycocotools provides a high performance implementation already. We have had to copy from master, this _dataset_wrapper.py because of a bug related to the handling of these masks that was fixed in master but not installable using pip yet.
🚀 The feature
The CocoDetection v2 transform wrapper attempts to decode the mask itself, but pycocotools provides a high performance implementation already. We have had to copy from master, this
_dataset_wrapper.py
because of a bug related to the handling of these masks that was fixed in master but not installable usingpip
yet.https://github.com/pytorch/vision/blob/main/torchvision/tv_tensors/_dataset_wrapper.py#L402
Seeing
torchvision.datasets.CocoDetection
hasself.coco
as aCOCO()
object, let's use it.Motivation, pitch
There have already been bugs related to this, and there's no need to reinvent the wheel. Instead, let's use the existing implementation.
Alternatives
No response
Additional context
No response
The text was updated successfully, but these errors were encountered: