Hello, I want to train my own data, there is only one category, and I encountered the following error during training:
start prefetching data...
shuffling indices...
Traceback (most recent call last):
File "train.py", line 61, in prefetch_data
data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data
return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection
db_ind = db.db_inds[k_ind]
IndexError: index 0 is out of bounds for axis 0 with size 0
Process Process-1:
Traceback (most recent call last):
File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 249, in _bootstrap
self.run()
File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 93, in run
self._target(*self._args, self._kwargs)
File "train.py", line 65, in prefetch_data
raise e
File "train.py", line 61, in prefetch_data
data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data
return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection
db_ind = db.db_inds[k_ind]
IndexError: index 0 is out of bounds for axis 0 with size 0
start prefetching data...
shuffling indices...
Traceback (most recent call last):
File "train.py", line 61, in prefetch_data
data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data
return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection
db_ind = db.db_inds[k_ind]
IndexError: index 0 is out of bounds for axis 0 with size 0
Process Process-2:
Traceback (most recent call last):
File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 249, in _bootstrap
self.run()
File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 93, in run
self._target(*self._args, *self._kwargs)
File "train.py", line 65, in prefetch_data
raise e
File "train.py", line 61, in prefetch_data
data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data
return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection
db_ind = db.db_inds[k_ind]
IndexError: index 0 is out of bounds for axis 0 with size 0
start prefetching data...
shuffling indices...
Traceback (most recent call last):
File "train.py", line 61, in prefetch_data
data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data
return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection
db_ind = db.db_inds[k_ind]
IndexError: index 0 is out of bounds for axis 0 with size 0
Process Process-3:
building model...
module_file: models.ExtremeNet
Traceback (most recent call last):
File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 249, in _bootstrap
self.run()
File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 93, in run
self._target(self._args, self._kwargs)
File "train.py", line 65, in prefetch_data
raise e
File "train.py", line 61, in prefetch_data
data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data
return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection
db_ind = db.db_inds[k_ind]
IndexError: index 0 is out of bounds for axis 0 with size 0
start prefetching data...
shuffling indices...
Traceback (most recent call last):
File "train.py", line 61, in prefetch_data
data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data
return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection
db_ind = db.db_inds[k_ind]
IndexError: index 0 is out of bounds for axis 0 with size 0
Process Process-4:
Traceback (most recent call last):
File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 249, in _bootstrap
self.run()
File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 93, in run
self._target(*self._args, **self._kwargs)
File "train.py", line 65, in prefetch_data
raise e
File "train.py", line 61, in prefetch_data
data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data
return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection
db_ind = db.db_inds[k_ind]
IndexError: index 0 is out of bounds for axis 0 with size 0
total parameters: 198531504
loading from pretrained model
loading from ./cache/ExtremeNet_250000.pkl
setting learning rate to: 0.00025
training start...
0%| | 0/250000 [00:00<?, ?it/s]
Hello, I want to train my own data, there is only one category, and I encountered the following error during training:
start prefetching data... shuffling indices... Traceback (most recent call last): File "train.py", line 61, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection db_ind = db.db_inds[k_ind] IndexError: index 0 is out of bounds for axis 0 with size 0 Process Process-1: Traceback (most recent call last): File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 249, in _bootstrap self.run() File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 93, in run self._target(*self._args, self._kwargs) File "train.py", line 65, in prefetch_data raise e File "train.py", line 61, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection db_ind = db.db_inds[k_ind] IndexError: index 0 is out of bounds for axis 0 with size 0 start prefetching data... shuffling indices... Traceback (most recent call last): File "train.py", line 61, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection db_ind = db.db_inds[k_ind] IndexError: index 0 is out of bounds for axis 0 with size 0 Process Process-2: Traceback (most recent call last): File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 249, in _bootstrap self.run() File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 93, in run self._target(*self._args, *self._kwargs) File "train.py", line 65, in prefetch_data raise e File "train.py", line 61, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection db_ind = db.db_inds[k_ind] IndexError: index 0 is out of bounds for axis 0 with size 0 start prefetching data... shuffling indices... Traceback (most recent call last): File "train.py", line 61, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection db_ind = db.db_inds[k_ind] IndexError: index 0 is out of bounds for axis 0 with size 0 Process Process-3: building model... module_file: models.ExtremeNet Traceback (most recent call last): File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 249, in _bootstrap self.run() File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 93, in run self._target(self._args, self._kwargs) File "train.py", line 65, in prefetch_data raise e File "train.py", line 61, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection db_ind = db.db_inds[k_ind] IndexError: index 0 is out of bounds for axis 0 with size 0 start prefetching data... shuffling indices... Traceback (most recent call last): File "train.py", line 61, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection db_ind = db.db_inds[k_ind] IndexError: index 0 is out of bounds for axis 0 with size 0 Process Process-4: Traceback (most recent call last): File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 249, in _bootstrap self.run() File "/home/wujiacheng/anaconda3/lib/python3.6/multiprocessing/process.py", line 93, in run self._target(*self._args, **self._kwargs) File "train.py", line 65, in prefetch_data raise e File "train.py", line 61, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug, debug=debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 248, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/home/wujiacheng/2020/ExtremeNet-text/sample/coco_extreme.py", line 85, in kp_detection db_ind = db.db_inds[k_ind] IndexError: index 0 is out of bounds for axis 0 with size 0 total parameters: 198531504 loading from pretrained model loading from ./cache/ExtremeNet_250000.pkl setting learning rate to: 0.00025 training start... 0%| | 0/250000 [00:00<?, ?it/s]
Sincerely hope to get your reply!