48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from pathlib import Path
|
|
|
|
from flandre.utils.RfMeta import RfFrameMeta, RfSequenceMeta
|
|
from flandre.utils.archive import to_zip
|
|
|
|
|
|
def f1():
|
|
# rr = RfSequenceMeta
|
|
tempdst = Path('/mnt/16T/private_dataset/ustemp')
|
|
arr = []
|
|
for (i,
|
|
file) in enumerate(Path('/mnt/16T/private_dataset/New Folder/steel-top/').glob('*pwi.bin')):
|
|
file = Path(file)
|
|
x, y, _ = file.name.split('_')
|
|
r = RfFrameMeta(encoder=0, robot_x=int(x) * 100, robot_y=int(y) * 100)
|
|
arr.append((file, r))
|
|
arr.sort(key=lambda item: (item[1].robot_y, item[1].robot_x))
|
|
|
|
arg = []
|
|
|
|
for i, item in enumerate(arr):
|
|
file, meta = item
|
|
meta.sequence_id = i
|
|
print(file.name, meta.sequence_id, meta.robot_x, meta.robot_y)
|
|
pic = file.with_suffix('.png')
|
|
farr = []
|
|
if pic.exists():
|
|
farr.append((pic, '.png'))
|
|
arg.append((file, meta, farr))
|
|
|
|
to_zip(arg, tempdst, Path('/mnt/16T/private_dataset/us/steel-top,U=30,M=PWI,S=(256 1502).zip'))
|
|
|
|
def f2():
|
|
tempdst = Path('/mnt/16T/private_dataset/ustemp')
|
|
src = Path('/mnt/16T/private_dataset/New Folder/T1,U=30,M=FMC,S=(256 256 1502)/')
|
|
arr = []
|
|
for (i,
|
|
file) in enumerate(Path('/mnt/16T/private_dataset/New Folder/steel-top/').glob('*pwi.bin')):
|
|
file = Path(file)
|
|
x, _ = file.name.split('_')
|
|
r = RfFrameMeta(encoder=0, robot_x=int(x) * 100, robot_y=int(y) * 100)
|
|
arr.append((file, r))
|
|
arr.sort(key=lambda item: (item[1].robot_y, item[1].robot_x))
|
|
print(arr)
|
|
|
|
if __name__ == '__main__':
|
|
f2()
|