Hello, thank you for your work.

When I train my own dataset, the following errors are reported in the verification phase. Can you help me?

in valid
in eval
Param Tag_threshold 12.0
Param detection_threshold 0.2
Traceback (most recent call last):
File "/mnt/Code/win_det_heatmaps/common_pytorch/net_modules.py", line 152, in evalNet
rectify = test_config.rectify, winScoreThres = test_config.windowT)
File "/mnt/Code/win_det_heatmaps/common_pytorch/group/tag_group.py", line 266, in group_corners_on_tags
grouped = parser.parse(np.float32(dets), np.float32(tags), idx, ratio, rectify) # shape=(num_of_windows, 4, 4)
File "/mnt/Code/win_det_heatmaps/common_pytorch/group/tag_group.py", line 242, in parse
re = self.calc(det, tag, idx)
File "/mnt/Code/win_det_heatmaps/common_pytorch/group/tag_group.py", line 178, in calc
val_k = [c_pts[:, 2, np.newaxis] for c_pts in coords_in_patch_with_score_id]
File "/mnt/Code/win_det_heatmaps/common_pytorch/group/tag_group.py", line 178, in
val_k = [c_pts[:, 2, np.newaxis] for c_pts in coords_in_patch_with_score_id]
IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "train.py", line 158, in
main()
File "train.py", line 135, in main
config.train.patch_width, config.train.patch_height, final_output_path)
File "/mnt/Code/win_det_heatmaps/common_pytorch/net_modules.py", line 155, in evalNet
assert 0, (n_s, e, os.path.basename(imdb_list[n_s]['image']))
AssertionError: (0, IndexError('too many indices for array: array is 1-dimensional, but 2 were indexed'), 'result_toushi_2282_6_4.jpg')