mirrors
/
mmselfsup
mirror of https://github.com/open-mmlab/mmselfsup.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Fix]: Change np.int to int (#636)

pull/637/head
Yuan Liu 2022-12-19 15:14:03 +08:00 committed by GitHub
parent fbe0ff6ec7
commit 1049791766
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmselfsup/datasets/samplers/deepcluster_sampler.py
View File

@ -101,7 +101,7 @@ class DeepClusterSampler(DefaultSampler):
l, size_per_label, replace=(len(l) <= size_per_label))) l, size_per_label, replace=(len(l) <= size_per_label)))
indices = np.array(indices) indices = np.array(indices)
np.random.shuffle(indices) np.random.shuffle(indices)
indices = indices[:N].astype(np.int).tolist() indices = indices[:N].astype(int).tolist()
# add extra samples to make it evenly divisible # add extra samples to make it evenly divisible
assert len(indices) <= self.total_size, \ assert len(indices) <= self.total_size, \

2
mmselfsup/datasets/transforms/processing.py
View File

@ -198,7 +198,7 @@ class BEiTMaskGenerator(BaseTransform):
Returns: Returns:
dict: Result dict with added key ``mask``. dict: Result dict with added key ``mask``.
""" """
mask = np.zeros(shape=self.get_shape(), dtype=np.int) mask = np.zeros(shape=self.get_shape(), dtype=int)
mask_count = 0 mask_count = 0
while mask_count != self.num_masking_patches: while mask_count != self.num_masking_patches:
max_mask_patches = self.num_masking_patches - mask_count max_mask_patches = self.num_masking_patches - mask_count