aim-uofa / AdelaiDet

AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.
https://git.io/AdelaiDet
Other
3.36k stars 643 forks source link

ABCNet Training Gives Error #113

Closed seecswajid closed 4 years ago

seecswajid commented 4 years ago
I am training ABCNet on my own data having 300 training images. I prepared my data using the scripts you have provided I am using your trained model on totaltext as a starting point. category #instances
text 300

However, after loading the model I got following error. !python tools/train_net.py \ --config-file configs/BAText/TotalText/attn_R_50.yaml \ --num-gpus 1 \ MODEL.WEIGHTS text_pretraining/attn_R_50/model_final.pth

Starting training from iteration 0 Traceback (most recent call last): File "tools/train_net.py", line 243, in args=(args,), File "/usr/local/lib/python3.6/dist-packages/detectron2/engine/launch.py", line 57, in launch main_func(args) File "tools/train_net.py", line 231, in main return trainer.train() File "tools/train_net.py", line 113, in train self.train_loop(self.start_iter, self.max_iter) File "tools/train_net.py", line 102, in train_loop self.run_step() File "/usr/local/lib/python3.6/dist-packages/detectron2/engine/train_loop.py", line 209, in run_step data = next(self._data_loader_iter) File "/usr/local/lib/python3.6/dist-packages/detectron2/data/common.py", line 142, in iter for d in self.dataset: File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 345, in next data = self._next_data() File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 856, in _next_data return self._process_data(data) File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 881, in _process_data data.reraise() File "/usr/local/lib/python3.6/dist-packages/torch/_utils.py", line 395, in reraise raise self.exc_type(msg) TypeError: Caught TypeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop data = fetcher.fetch(index) File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/lib/python3.6/dist-packages/detectron2/data/common.py", line 41, in getitem data = self._map_func(self._dataset[cur_idx]) File "/usr/local/lib/python3.6/dist-packages/detectron2/utils/serialize.py", line 23, in call return self._obj(args, **kwargs) File "/content/drive/My Drive/Colab Notebooks/Bezier_detectron/AdelaiDet/adet/data/dataset_mapper.py", line 88, in call image = crop_tfm.apply_image(image) File "/usr/local/lib/python3.6/dist-packages/fvcore/transforms/transform.py", line 598, in apply_image return img[self.y0 : self.y0 + self.h, self.x0 : self.x0 + self.w] TypeError: slice indices must be integers or None or have an index method

Yuliang-Liu commented 4 years ago

@seecswajid

  1. Can you train the example we provided without changing anything?
  2. slice indices must be integers. Can you show the value of the indices.
seecswajid commented 4 years ago

1) I tried training with example you provided but it give me same error. 2) These values are from another function and I am unable to print them.