Skip to content

UnboundLocalError: local variable 'current' referenced before assignment #55

@chenzhy223

Description

@chenzhy223

当我使用数据集GOT-10k重复训练TCTrack的时候,加载完预训练模型之后会报错:
Traceback (most recent call last):
File "d:/code/trackers/TCTrack/tools/train_tctrack.py", line 308, in
main()
File "d:/code/trackers/TCTrack/tools/train_tctrack.py", line 303, in main
train(train_loader, dist_model, optimizer, lr_scheduler, tb_writer)
File "d:/code/trackers/TCTrack/tools/train_tctrack.py", line 149, in train
for idx, data in enumerate(train_loader):
File "C:\Users*.conda\envs\TCTrack\lib\site-packages\torch\utils\data\dataloader.py", line 634, in next
data = self._next_data()
File "C:\Users*
.conda\envs\TCTrack\lib\site-packages\torch\utils\data\dataloader.py", line 1346, in _next_data
return self._process_data(data)
File "C:\Users*.conda\envs\TCTrack\lib\site-packages\torch\utils\data\dataloader.py", line 1372, in _process_data
data.reraise()
File "C:\Users*
.conda\envs\TCTrack\lib\site-packages\torch_utils.py", line 644, in reraise
raise exception
UnboundLocalError: Caught UnboundLocalError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\Users*.conda\envs\TCTrack\lib\site-packages\torch\utils\data_utils\worker.py", line 308, in _worker_loop
data = fetcher.fetch(index)
File "C:\Users*
.conda\envs\TCTrack\lib\site-packages\torch\utils\data_utils\fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users******.conda\envs\TCTrack\lib\site-packages\torch\utils\data_utils\fetch.py", line 51, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\code\trackers\TCTrack\tools..\pysot\datasets\dataset.py", line 296, in getitem
current_templateimage = cv2.imread(current[0][0])
UnboundLocalError: local variable 'current' referenced before assignment

根据报错,定位到文件pysot\datasets\dataset.py中的

Image
会存在neg为true的情况,导致current没有赋值。

请问这个要怎么解决呢?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions