Dear @dusty_nv
seems I have a bounding box with zero inside during trial 1
mode = random.choice(self.sample_options)
[[ 994. 419. 1043. 474.]
[1399. 357. 1476. 405.]
[1230. 341. 1305. 411.]
[ 908. 312. 962. 368.]]
4
[[136. 448. 182. 495.]
[188. 456. 258. 528.]
[150. 387. 216. 443.]
[ 64. 296. 117. 354.]
[434. 392. 566. 484.]
[588. 284. 674. 378.]]
6
[]
0
[[ 87. 0. 128. 37.]]
1
[[ 97. 154. 218. 242.]]
1
[[19. 58. 58. 97.]]
1
Executing the python3 train_ssd.py
another time, I receive the following output
mode = random.choice(self.sample_options)
[[1717. 750. 1791. 812.]
[1644. 773. 1692. 829.]
[1520. 630. 1582. 702.]]
3
[[ 449. 1557. 516. 1611.]
[ 621. 1423. 695. 1496.]
[ 198. 1601. 279. 1664.]
[ 156. 1530. 243. 1591.]
[ 153. 1589. 204. 1643.]]
5
[[2366. 571. 2415. 626.]
[1933. 509. 2010. 557.]
[2104. 493. 2179. 563.]
[2447. 464. 2501. 520.]]
4
[[557. 4. 629. 69.]]
1
[[118. 480. 220. 548.]
[509. 241. 554. 296.]
[540. 311. 576. 345.]
[627. 272. 696. 317.]]
4
/home/aimobile/.local/lib/python3.6/site-packages/torch/nn/_reduction.py:44: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
Traceback (most recent call last):
File "train_ssd.py", line 343, in <module>
device=DEVICE, debug_steps=args.debug_steps, epoch=epoch)
File "train_ssd.py", line 113, in train
for i, data in enumerate(loader):
File "/home/aimobile/.local/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 363, in __next__
data = self._next_data()
File "/home/aimobile/.local/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 971, in _next_data
return self._process_data(data)
File "/home/aimobile/.local/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1014, in _process_data
data.reraise()
File "/home/aimobile/.local/lib/python3.6/site-packages/torch/_utils.py", line 395, in reraise
raise self.exc_type(msg)
IndexError: Caught IndexError in DataLoader worker process 1.
Original Traceback (most recent call last):
File "/home/aimobile/.local/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 185, in _worker_loop
data = fetcher.fetch(index)
File "/home/aimobile/.local/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/aimobile/.local/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/aimobile/.local/lib/python3.6/site-packages/torch/utils/data/dataset.py", line 207, in __getitem__
return self.datasets[dataset_idx][sample_idx]
File "/home/aimobile/jetson-inference/python/training/detection/ssd/vision/datasets/voc_dataset.py", line 64, in __getitem__
image, boxes, labels = self.transform(image, boxes, labels)
File "/home/aimobile/jetson-inference/python/training/detection/ssd/vision/ssd/data_preprocessing.py", line 34, in __call__
return self.augment(img, boxes, labels)
File "/home/aimobile/jetson-inference/python/training/detection/ssd/vision/transforms/transforms.py", line 55, in __call__
img, boxes, labels = t(img, boxes, labels)
File "/home/aimobile/jetson-inference/python/training/detection/ssd/vision/transforms/transforms.py", line 345, in __call__
boxes[:, :2] += (int(left), int(top))
IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed