Update VAD_test

Thinklab-SJTU · Jun 25, 2024 · 8dd2cfc · 8dd2cfc
1 parent 3e79d99
commit 8dd2cfc
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/adzoo/vad/apis/test.py b/adzoo/vad/apis/test.py
@@ -77,14 +77,6 @@ def custom_multi_gpu_test(model, data_loader, tmpdir=None, gpu_collect=False):
             else:
                 batch_size = len(result)
                 bbox_results.extend(result)
-
-            if i>150:
-                break
-
-            #if isinstance(result[0], tuple):
-            #    assert False, 'this code is for instance segmentation, which our code will not utilize.'
-            #    result = [(bbox_results, encode_mask_results(mask_results))
-            #              for bbox_results, mask_results in result]
         if rank == 0:
 
             for _ in range(batch_size * world_size):