Skip to content

Commit 1d3a367

Browse files
authored
Update generate_pic_eigenvectors.py
Removing redundant lines of code.
1 parent f2efe98 commit 1d3a367

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

generate_pic_eigenvectors.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,7 @@ def generate_pic_eigenvectors(args):
1414
features_path = DATA_PATH / args.dataset / args.features
1515
scale_factor = np.ceil(int(args.features.split('_')[-1]) / int(args.features.split('_')[-3])).astype(np.int32)
1616

17-
dino_masks = DATA_PATH / args.dataset / f'pic_masks_f{args.flow}_b{args.beta}_w{args.flow_weight}_s{scale_factor}'
1817
pic_eigenvectors = DATA_PATH / args.dataset / f'pic_eigenvectors_f{args.flow}_b{args.beta}_w{args.flow_weight}_s{scale_factor}'
19-
dino_masks.mkdir(exist_ok=True)
2018
pic_eigenvectors.mkdir(exist_ok=True)
2119

2220
(DATA_PATH / args.dataset / f'pic_config_f{args.flow}_b{args.beta}_w{args.flow_weight}_s{scale_factor}.yaml').write_text(yaml.dump(args))
@@ -29,9 +27,7 @@ def generate_pic_eigenvectors(args):
2927
for obj_class in sorted(features_path.iterdir()):
3028
print(f'Working on class: {obj_class.name}')
3129

32-
dino_masks_class = dino_masks / obj_class.name
3330
pic_eigenvectors_class = pic_eigenvectors / obj_class.name
34-
dino_masks_class.mkdir(exist_ok=True)
3531
pic_eigenvectors_class.mkdir(exist_ok=True)
3632

3733
n_features = len(list(obj_class.glob('*.*')))
@@ -106,4 +102,4 @@ def generate_pic_eigenvectors(args):
106102
parser.add_argument('--use-gpu', action='store_true', help="use flow for mask generation")
107103
args = parser.parse_args()
108104

109-
generate_pic_eigenvectors(args)
105+
generate_pic_eigenvectors(args)

0 commit comments

Comments
 (0)