47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
import os
|
|
|
|
if __name__ == "__main__":
|
|
trg_dir = "../datasets/pieces/unified/train/labels"
|
|
src_dir = "../datasets/pieces/khalid/train/labels"
|
|
|
|
trg_labels = [
|
|
'w_pawn','w_knight','w_bishop','w_rook','w_queen','w_king',
|
|
'b_pawn','b_knight','b_bishop','b_rook','b_queen','b_king'
|
|
]
|
|
|
|
src_labels = [
|
|
'b_bishop', 'b_king', 'b_knight', 'b_queen', 'b_rook', 'b_pawn',
|
|
'w_bishop', 'w_king', 'w_knight', 'w_queen', 'w_rook', 'w_pawn'
|
|
]
|
|
|
|
trg_files = os.listdir(trg_dir)
|
|
src_files = os.listdir(src_dir)
|
|
|
|
for src_file in src_files:
|
|
trg_file = os.path.abspath(os.path.join(trg_dir, src_file))
|
|
src_file = os.path.abspath(os.path.join(src_dir, src_file))
|
|
|
|
trg_lines = []
|
|
src_lines = []
|
|
|
|
with open(src_file, "r") as f:
|
|
src_lines = f.readlines().copy()
|
|
|
|
with open(trg_file, "r") as f:
|
|
trg_lines = f.readlines().copy()
|
|
|
|
for i in range(0, len(trg_lines)):
|
|
trg_line = trg_lines[i]
|
|
src_line = src_lines[i]
|
|
|
|
trg_label_index = int(trg_line.strip().split(" ")[0])
|
|
src_label_index = int(src_line.strip().split(" ")[0])
|
|
|
|
trg_label_value = trg_labels[trg_label_index]
|
|
src_label_value = src_labels[src_label_index]
|
|
|
|
if trg_label_value != src_label_value :
|
|
print(f"Error detected in {trg_file} at line {i}.\n"
|
|
f"==> Index {trg_label_index} points to {trg_label_value} instead of {src_label_index}:{src_label_value}")
|
|
|
|
print("Detection terminated.") |