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=
|
| 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):
|