modify: unsup student get

Seeed-Studio · Nov 24, 2023 · 587ecee · 587ecee
1 parent 767067e
commit 587ecee
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/sscma/datasets/transforms/wrappers.py b/sscma/datasets/transforms/wrappers.py
@@ -18,11 +18,7 @@ def transform(self, results: Dict) -> Optional[Union[Dict, Tuple[List, List]]]:
         for branch, pipeline in self.branch_pipelines.items():
             branch_results = pipeline(copy.deepcopy(results))
             if branch == 'unsup_teacher':
-                teach = branch_results['inputs'].permute(1, 2, 0).cpu().numpy()
                 results['img'] = branch_results['inputs'].permute(1, 2, 0).cpu().numpy()
-
-            elif branch == 'unsup_student':
-                stu = branch_results['inputs'].permute(1, 2, 0).cpu().numpy()
             # If one branch pipeline returns None,
             # it will sample another data from dataset.
             if branch_results is None: