(self, primact_types, category_types, data_features, buffer_max_num, \
abs_thres=0.01, rel_thres=0.5, dp_thres=0.5, img_size=224, \
no_true_false_equal=False, no_neg_dir_data=False, only_true_data=False)
| 19 | class SAPIENVisionDataset(data.Dataset): |
| 20 | |
| 21 | def __init__(self, primact_types, category_types, data_features, buffer_max_num, \ |
| 22 | abs_thres=0.01, rel_thres=0.5, dp_thres=0.5, img_size=224, \ |
| 23 | no_true_false_equal=False, no_neg_dir_data=False, only_true_data=False): |
| 24 | self.primact_types = primact_types |
| 25 | self.category_types = category_types |
| 26 | |
| 27 | self.buffer_max_num = buffer_max_num |
| 28 | self.img_size = img_size |
| 29 | self.abs_thres = abs_thres |
| 30 | self.rel_thres = rel_thres |
| 31 | self.dp_thres = dp_thres |
| 32 | self.no_true_false_equal = no_true_false_equal |
| 33 | self.no_neg_dir_data = no_neg_dir_data |
| 34 | self.only_true_data = only_true_data |
| 35 | |
| 36 | # data buffer |
| 37 | self.true_data = dict() |
| 38 | self.false_data = dict() |
| 39 | for primact_type in primact_types: |
| 40 | self.true_data[primact_type] = [] |
| 41 | self.false_data[primact_type] = [] |
| 42 | |
| 43 | # data features |
| 44 | self.data_features = data_features |
| 45 | |
| 46 | def load_data(self, data_list): |
| 47 | bar = ProgressBar() |
nothing calls this directly
no outgoing calls
no test coverage detected