bulatko commited on
Commit
cb95dfb
·
1 Parent(s): c46a560

Fix MSDeformAttn: update deprecated PyTorch API for PyTorch 2.x compatibility

Browse files
MaskClustering/semantics/get_open-voc_features.py CHANGED
@@ -129,7 +129,7 @@ def main():
129
  segmentation_path_list.append(segmentation_path)
130
  feature_dict[seq_name] = {}
131
 
132
- dataloader = DataLoader(CroppedImageDataset(seq_name_list, frame_id_list, mask_id_list, rgb_path_list, segmentation_path_list, preprocess), batch_size=64, shuffle=False, num_workers=16)
133
 
134
  print('[INFO] extracting features')
135
  for images, seq_names, frame_ids, mask_ids in tqdm(dataloader):
 
129
  segmentation_path_list.append(segmentation_path)
130
  feature_dict[seq_name] = {}
131
 
132
+ dataloader = DataLoader(CroppedImageDataset(seq_name_list, frame_id_list, mask_id_list, rgb_path_list, segmentation_path_list, preprocess), batch_size=4, shuffle=False, num_workers=16)
133
 
134
  print('[INFO] extracting features')
135
  for images, seq_names, frame_ids, mask_ids in tqdm(dataloader):