x-lai commited on
Commit
3caeb35
·
1 Parent(s): 7c90bfa

add print samples of reason_seg_dataset

Browse files

Former-commit-id: 78042e4cdb0cc7dd803fbe1fa9651db3febf0580

Files changed (1) hide show
  1. utils/reason_seg_dataset.py +2 -0
utils/reason_seg_dataset.py CHANGED
@@ -74,6 +74,8 @@ class ReasonSegDataset(torch.utils.data.Dataset):
74
  jsons = [path.replace(".jpg", ".json") for path in images]
75
  self.reason_seg_data = (images, jsons)
76
 
 
 
77
  if explanatory != -1:
78
  self.explanatory_question_list = EXPLANATORY_QUESTION_LIST
79
  self.img_to_explanation = {}
 
74
  jsons = [path.replace(".jpg", ".json") for path in images]
75
  self.reason_seg_data = (images, jsons)
76
 
77
+ print("number of reason_seg samples: ", len(images))
78
+
79
  if explanatory != -1:
80
  self.explanatory_question_list = EXPLANATORY_QUESTION_LIST
81
  self.img_to_explanation = {}