Fix inverted probability logic in RandomExpand.__call__

Author: Pantelis

rfdetr/datasets/transforms.py

@@ -391,7 +391,7 @@ def __init__(self, ratio=4., prob=0.5, fill_value=(127.5, 127.5, 127.5)):
         self.fill_value = fill_value
 
     def __call__(self, img, target):
-        if np.random.uniform(0., 1.) < self.prob:
+        if np.random.uniform(0., 1.) >= self.prob:
             return img, target
 
         height, width = img.shape[:2]