mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
d7c1294d9e
Summary: move div255 to gpu add read/write numpy ndarray, which will make the comparison between torch and trt results more easily. Reviewed By: l1aoxingyu
16 lines
271 B
Python
Executable File
16 lines
271 B
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys
|
|
from numpy import load
|
|
from scipy.io import savemat
|
|
|
|
assert len(sys.argv) > 1
|
|
|
|
files = sys.argv[1:]
|
|
|
|
for f in files:
|
|
data = load(f)
|
|
fn = f.replace('.npz','')
|
|
fn = fn.replace('.','_') #matlab cant handle dots
|
|
savemat(fn,data)
|