关于PicklingError的问题

[Jonyond-lin]

您好，我在复现的过程中碰到了如下问题

(CPT) [root@ts-8789b3171f704463b2163b0583223e52-launcher ~/jonyond/CentripetalText-main]# CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py config/ct/ct_r18_tt.py { "model": { "type": "CT", "backbone": { "type": "resnet18", "pretrained": true }, "neck": { "type": "FPEM_v1", "in_channels": [ 64, 128, 256, 512 ], "out_channels": 128 }, "detection_head": { "type": "CT_Head", "in_channels": 512, "hidden_dim": 128, "num_classes": 3, "loss_kernel": { "type": "DiceLoss", "loss_weight": 1.0 }, "loss_loc": { "type": "SmoothL1Loss", "beta": 0.1, "loss_weight": 0.05 } } }, "data": { "batch_size": 16, "train": { "type": "CT_TT", "split": "train", "is_transform": true, "img_size": 640, "short_size": 640, "kernel_scale": 0.7, "read_type": "cv2" }, "test": { "type": "CT_TT", "split": "test", "short_size": 640, "read_type": "cv2" } }, "train_cfg": { "lr": 0.001, "schedule": "polylr", "epoch": 600, "optimizer": "Adam" }, "test_cfg": { "min_score": 0.88, "min_area": 16, "bbox_type": "poly", "result_path": "outputs/submit_tt/" } } Checkpoint path: checkpoints/ct_r18_tt.

Epoch: [1 | 600] Traceback (most recent call last): File "/root/anaconda3/envs/CPT/lib/python3.6/multiprocessing/queues.py", line 234, in _feed obj = _ForkingPickler.dumps(obj) File "/root/anaconda3/envs/CPT/lib/python3.6/multiprocessing/reduction.py", line 51, in dumps cls(buf, protocol).dump(obj) _pickle.PicklingError: Can't pickle <class 'cPolygon.Error'>: import of module 'cPolygon' failed /root/anaconda3/envs/CPT/lib/python3.6/site-packages/torch/nn/functional.py:2539: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details. "See the documentation of nn.Upsample for details.".format(mode)) /root/anaconda3/envs/CPT/lib/python3.6/site-packages/torch/nn/functional.py:2457: UserWarning: nn.functional.upsample is deprecated. Use nn.functional.interpolate instead. warnings.warn("nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.") (1/78) LR: 0.001000 | Batch: 32.551s | Total: 0min | ETA: 42min | Loss: 1.303 | Loss(kernel/loc): 0.812/0.491 | IoU(text/kernel): 0.893907/0.426699

请问这种情况如何解决呢？cPolygon好像根本没有这个包，我在网上都找不到

[shengtao96]

请确认是否安装Polygon3。

推荐按照“Recommended environment”中的配置搭建训练环境。

[Jonyond-lin]

请确认是否安装Polygon3。

推荐按照“Recommended environment”中的配置搭建训练环境。

(CPT) [root@ts-8789b3171f704463b2163b0583223e52-launcher ~]# pip install Polygon3 Looking in indexes: http://mirrors.cloud.tencent.com/pypi/simple Requirement already satisfied: Polygon3 in ./anaconda3/envs/CPT/lib/python3.6/site-packages (3.0.9.1) WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv (CPT) [root@ts-8789b3171f704463b2163b0583223e52-launcher ~]# pip install Polygon3 Looking in indexes: http://mirrors.cloud.tencent.com/pypi/simple Requirement already satisfied: Polygon3 in ./anaconda3/envs/CPT/lib/python3.6/site-packages (3.0.9.1)

环境安装基本上是按照Recommended environment来的，但还是输出了 Epoch: [1 | 600] Traceback (most recent call last): File "/root/anaconda3/envs/CPT/lib/python3.6/multiprocessing/queues.py", line 234, in _feed obj = _ForkingPickler.dumps(obj) File "/root/anaconda3/envs/CPT/lib/python3.6/multiprocessing/reduction.py", line 51, in dumps cls(buf, protocol).dump(obj) _pickle.PicklingError: Can't pickle <class 'cPolygon.Error'>: import of module 'cPolygon' failed /root/anaconda3/envs/CPT/lib/python3.6/site-packages/torch/nn/functional.py:2457: UserWarning: nn.functional.upsample is deprecated. Use nn.functional.interpolate instead. warnings.warn("nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.") /root/anaconda3/envs/CPT/lib/python3.6/site-packages/torch/nn/functional.py:2539: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details. "See the documentation of nn.Upsample for details.".format(mode)) (1/78) LR: 0.001000 | Batch: 21.952s | Total: 0min | ETA: 29min | Loss: 1.303 | Loss(kernel/loc): 0.812/0.491 | IoU(text/kernel): 0.893907/0.426699 并且一直在这这个地方，这个是正常情况吗，还是说训练的时长就是比较久？

[guoli188]

我也遇到了这个问题，请问解决了吗？谢谢

[Jonyond-lin]

还是没解决，请问您那边解决了吗？