Some code bugs

[RoyAmoyal]

Hey, I am trying to run the code with your pretrained weights. (without training it by myself)

How to run the code just for testing/visualization?

Following your guide:
when running python3 main.py --root_dataset="datasets/" --mode=test --demo_mode=True
I am getting the next error:
FileNotFoundError: [Errno 2] No such file or directory: 'datasets/LINEMOD/ape/Split/train.txt'
or 'datasets/LINEMOD/ape/Split/val.txt'

that because there is not Split directory (the ape/Split/ doesn't exist) but instead the files inside the ape/ folder

Can you please try to edit the guide for us to run your code from scratch?

Thanks for your great work!

[lin-fangzhou]

Have you solved this problem?

[ignorantimt]

I wrote a Python script to solve this problem.

import os
import re

linemod_cls_names = ['ape','benchvise','cam','can','cat','driller','duck','eggbox','glue','holepuncher','iron','lamp','phone']

pattern = r'\d+'

for cls_name in linemod_cls_names:
    folder_path = os.path.join('datasets','LINEMOD',cls_name)
    val_txt_path = os.path.join(folder_path,'val.txt')
    output_path = os.path.join(folder_path,'Split','val.txt')

    with open(val_txt_path,'r') as val_file:
        lines = val_file.readlines()

    cleaned_lines = []
    for line in lines:
        match = re.search(pattern,line)
        if match:
            cleaned_lines.append(match.group())

    with open(output_path, 'w') as split_val_file:
        split_val_file.write('\n'.join(cleaned_lines))

print('done!')

---

https://github.com/aaronWool/rcvpose/blob/c1490bf79450093fdb23e41b72e80e339d192942/AccumulatorSpace.py#L499C26-L499C26
You can also edit original codes.

test_list = open(opts.root_dataset + "LINEMOD/"+class_name+"/" +"val.txt","r").readlines()