@@ -14,9 +14,7 @@ def generate_pic_eigenvectors(args):
14
14
features_path = DATA_PATH / args .dataset / args .features
15
15
scale_factor = np .ceil (int (args .features .split ('_' )[- 1 ]) / int (args .features .split ('_' )[- 3 ])).astype (np .int32 )
16
16
17
- dino_masks = DATA_PATH / args .dataset / f'pic_masks_f{ args .flow } _b{ args .beta } _w{ args .flow_weight } _s{ scale_factor } '
18
17
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 )
20
18
pic_eigenvectors .mkdir (exist_ok = True )
21
19
22
20
(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):
29
27
for obj_class in sorted (features_path .iterdir ()):
30
28
print (f'Working on class: { obj_class .name } ' )
31
29
32
- dino_masks_class = dino_masks / obj_class .name
33
30
pic_eigenvectors_class = pic_eigenvectors / obj_class .name
34
- dino_masks_class .mkdir (exist_ok = True )
35
31
pic_eigenvectors_class .mkdir (exist_ok = True )
36
32
37
33
n_features = len (list (obj_class .glob ('*.*' )))
@@ -106,4 +102,4 @@ def generate_pic_eigenvectors(args):
106
102
parser .add_argument ('--use-gpu' , action = 'store_true' , help = "use flow for mask generation" )
107
103
args = parser .parse_args ()
108
104
109
- generate_pic_eigenvectors (args )
105
+ generate_pic_eigenvectors (args )
0 commit comments