Update pred_zy3bh_tlcnetU_tlc.py

Author: YinxiaCao

pred_zy3bh_tlcnetU_tlc.py

@@ -26,7 +26,7 @@ def main():
     data_path = r'sample'
     batch_size = 16
     # Load dataset
-    testimg, testlab, nameid = dataloaderbh_testall(data_path)
+    testimg, testlab, nameid = dataloaderbh_testall(data_path,[0,0,1])
 
     testdataloader = torch.utils.data.DataLoader(
         myImageFloder_tlc(testimg, testlab, num=16),