Update extract_feature_print.py
This commit is contained in:
parent
66039595c9
commit
a75619c453
|
@ -80,9 +80,11 @@ else:
|
||||||
feats = model.final_proj(logits[0])
|
feats = model.final_proj(logits[0])
|
||||||
|
|
||||||
feats = feats.squeeze(0).float().cpu().numpy()
|
feats = feats.squeeze(0).float().cpu().numpy()
|
||||||
# feats = np.repeat(feats, 2,0) # 20ms -> 10ms
|
if(np.isnan(feats).sum()==0):
|
||||||
np.save(out_path, feats, allow_pickle=False)
|
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))
|
if (idx % n == 0):printt("now-%s,all-%s,%s,%s"%(len(todo),idx,file,feats.shape))
|
||||||
except:
|
except:
|
||||||
printt(traceback.format_exc())
|
printt(traceback.format_exc())
|
||||||
printt("all-feature-done")
|
printt("all-feature-done")
|
||||||
|
|
Loading…
Reference in New Issue