Hi @MhLiao
I encountered an error when i try to finetuning pretrain model. Can you help me solving this error?
Thanks in advance.
Traceback (most recent call last):
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/tools/train_net.py", line 173, in
main()
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/tools/train_net.py", line 166, in main
model = train(cfg, args.local_rank, args.distributed)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/tools/train_net.py", line 76, in train
cfg,
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/engine/trainer.py", line 66, in do_train
loss_dict = model(images, targets)
File "/media/datngo/DATA/Projects/Cores/DB/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, kwargs)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/modeling/detector/generalized_rcnn.py", line 52, in forward
x, result, detector_losses = self.roi_heads(features, proposals, targets)
File "/media/datngo/DATA/Projects/Cores/DB/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, *kwargs)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/modeling/roi_heads/roi_heads.py", line 36, in forward
x, detections, loss_mask = self.mask(mask_features, detections, targets)
File "/media/datngo/DATA/Projects/Cores/DB/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(input, kwargs)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/modeling/roi_heads/mask_head/mask_head.py", line 193, in forward
mask_targets, char_mask_targets, char_mask_weights, decoder_targets, word_targets = self.prepare_targets(proposals, targets)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/modeling/roi_heads/mask_head/mask_head.py", line 156, in prepare_targets
segmentation_masks, char_segmentation_masks, positive_proposals, self.discretization_size
ValueError: not enough values to unpack (expected 5, got 4)
Hi @MhLiao I encountered an error when i try to finetuning pretrain model. Can you help me solving this error? Thanks in advance.
Traceback (most recent call last): File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/tools/train_net.py", line 173, in
main()
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/tools/train_net.py", line 166, in main
model = train(cfg, args.local_rank, args.distributed)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/tools/train_net.py", line 76, in train
cfg,
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/engine/trainer.py", line 66, in do_train
loss_dict = model(images, targets)
File "/media/datngo/DATA/Projects/Cores/DB/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, kwargs)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/modeling/detector/generalized_rcnn.py", line 52, in forward
x, result, detector_losses = self.roi_heads(features, proposals, targets)
File "/media/datngo/DATA/Projects/Cores/DB/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, *kwargs)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/modeling/roi_heads/roi_heads.py", line 36, in forward
x, detections, loss_mask = self.mask(mask_features, detections, targets)
File "/media/datngo/DATA/Projects/Cores/DB/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(input, kwargs)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/modeling/roi_heads/mask_head/mask_head.py", line 193, in forward
mask_targets, char_mask_targets, char_mask_weights, decoder_targets, word_targets = self.prepare_targets(proposals, targets)
File "/media/datngo/DATA/Projects/Cores/MaskTextSpotter/maskrcnn_benchmark/modeling/roi_heads/mask_head/mask_head.py", line 156, in prepare_targets
segmentation_masks, char_segmentation_masks, positive_proposals, self.discretization_size
ValueError: not enough values to unpack (expected 5, got 4)