Update extract_feature_print.py

This commit is contained in:
liujing04 2023-04-09 15:31:38 +00:00 committed by GitHub
parent 66039595c9
commit a75619c453
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions

View File

@ -80,9 +80,11 @@ else:
feats = model.final_proj(logits[0])
feats = feats.squeeze(0).float().cpu().numpy()
# feats = np.repeat(feats, 2,0) # 20ms -> 10ms
np.save(out_path, feats, allow_pickle=False)
if(np.isnan(feats).sum()==0):
np.save(out_path, feats, allow_pickle=False)
else:
printt("%s-contains nan"%file)
if (idx % n == 0):printt("now-%s,all-%s,%s,%s"%(len(todo),idx,file,feats.shape))
except:
printt(traceback.format_exc())
printt("all-feature-done")
printt("all-feature-done")