-
Notifications
You must be signed in to change notification settings - Fork 40
Description
当我使用数据集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中的
请问这个要怎么解决呢?
