-

Ainimal · Jan 23, 2023 · 17ae65c · 17ae65c
1 parent fc7d08d
commit 17ae65c
Show file tree

Hide file tree

Showing 10 changed files with 1 addition and 1 deletion.
diff --git a/demo/multi_label/Demo_CVPR2016_MultiLabel_CNNRNN.py b/demo/multi_label/Demo_CVPR2016_MultiLabel_CNNRNN.py
@@ -121,7 +121,7 @@ def inference(self, image, states=None):
  image_2D_tensor = (torch.tensor(np.stack([case['img_2D'].astype(np.float32) for case in dataset], 0))).permute(0, 3, 1, 2)
  image_3D_tensor = (torch.tensor(np.stack([case['img_3D'].astype(np.float32) for case in dataset], 0))).permute(0, 4, 1, 2, 3)
 
- # ML-GCN only support binary labels
+ # CNNRNN only support binary labels
  label_category_dict['weather'] = 2 # 5 → 2, force binary classification
 
 

diff --git a/wama_modules/__pycache__/BaseModule.cpython-38.pyc b/wama_modules/__pycache__/BaseModule.cpython-38.pyc
diff --git a/wama_modules/__pycache__/Decoder.cpython-38.pyc b/wama_modules/__pycache__/Decoder.cpython-38.pyc
diff --git a/wama_modules/__pycache__/Encoder.cpython-38.pyc b/wama_modules/__pycache__/Encoder.cpython-38.pyc
diff --git a/wama_modules/__pycache__/Head.cpython-38.pyc b/wama_modules/__pycache__/Head.cpython-38.pyc
diff --git a/wama_modules/__pycache__/Neck.cpython-38.pyc b/wama_modules/__pycache__/Neck.cpython-38.pyc
diff --git a/wama_modules/__pycache__/PositionEmbedding.cpython-38.pyc b/wama_modules/__pycache__/PositionEmbedding.cpython-38.pyc
diff --git a/wama_modules/__pycache__/Transformer.cpython-38.pyc b/wama_modules/__pycache__/Transformer.cpython-38.pyc
diff --git a/wama_modules/__pycache__/__init__.cpython-38.pyc b/wama_modules/__pycache__/__init__.cpython-38.pyc
diff --git a/wama_modules/__pycache__/utils.cpython-38.pyc b/wama_modules/__pycache__/utils.cpython-38.pyc