xingyizhou / ExtremeNet

Bottom-up Object Detection by Grouping Extreme and Center Points
BSD 3-Clause "New" or "Revised" License
1.03k stars 174 forks source link

Train my own data errors,Hope to get your help, Sincere thanks! #38

Open xdwjc opened 4 years ago

xdwjc commented 4 years ago

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!