MCPcopy Create free account
hub / github.com/bbaaii/DreamDiffusion / remove_repeats

Function remove_repeats

code/dataset.py:194–207  ·  view source on GitHub ↗
(fmri, img_lb)

Source from the content-addressed store, hash-verified

192 return self.transform(self.x[index]), self.transform(self.y[index])
193
194def remove_repeats(fmri, img_lb):
195 assert len(fmri) == len(img_lb), 'len error'
196 fmri_dict = {}
197 for f, lb in zip(fmri, img_lb):
198 if lb in fmri_dict.keys():
199 fmri_dict[lb].append(f)
200 else:
201 fmri_dict[lb] = [f]
202 lbs = []
203 fmris = []
204 for k, v in fmri_dict.items():
205 lbs.append(k)
206 fmris.append(np.mean(np.stack(v), axis=0))
207 return np.stack(fmris), lbs
208
209
210def list_get_all_index(list, value):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected