Spaces:
Paused
Paused
x-lai
commited on
Commit
·
3caeb35
1
Parent(s):
7c90bfa
add print samples of reason_seg_dataset
Browse filesFormer-commit-id: 78042e4cdb0cc7dd803fbe1fa9651db3febf0580
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 = {}
|