Disable warnings

Author: NicolasHug

references/detection/coco_utils.py

@@ -6,7 +6,7 @@
 
 from pycocotools import mask as coco_mask
 from pycocotools.coco import COCO
-from torchvision.datapoints import wrap_dataset_for_transforms_v2
+from torchvision.datasets import wrap_dataset_for_transforms_v2
 
 
 def convert_coco_poly_to_mask(segmentations, height, width):

references/detection/presets.py

@@ -2,6 +2,8 @@
 
 import torch
 import transforms as reference_transforms
+import torchvision
+torchvision.disable_beta_transforms_warning()
 from torchvision import datapoints
 import torchvision.transforms.v2 as T
 

references/detection/train.py

@@ -164,6 +164,7 @@ def get_args_parser(add_help=True):
 
 
 def main(args):
+
     if args.output_dir:
         utils.mkdir(args.output_dir)