加上参数vis进行结果可视化出现问题

[120L021304]

Load checkpoint from pretrained/13_Faster_RCNN.pt. [INFO] 2023-07-06 15:43:05 ==================Configs================== [INFO] 2023-07-06 15:43:05 MODEL: [INFO] 2023-07-06 15:43:05 NAME: Faster_RCNN [INFO] 2023-07-06 15:43:05 BACKBONE: resnet50 [INFO] 2023-07-06 15:43:05 [INFO] 2023-07-06 15:43:05 DATA: [INFO] 2023-07-06 15:43:05 DATASET: mydata [INFO] 2023-07-06 15:43:05 TRANSFORM: frcnn [INFO] 2023-07-06 15:43:05 SCALE: [800, 1333] [INFO] 2023-07-06 15:43:05 OPTIMIZE: [INFO] 2023-07-06 15:43:05 OPTIMIZER: sgd [INFO] 2023-07-06 15:43:05 BASE_LR: 0.001 [INFO] 2023-07-06 15:43:05 SCHEDULER: 1x [INFO] 2023-07-06 15:43:05 BATCH_SIZE: 2 [INFO] 2023-07-06 15:43:05 TEST: [INFO] 2023-07-06 15:43:05 NMS_THRESH : 0.5 [INFO] 2023-07-06 15:43:05 CONF_THRESH: 0.05 [INFO] 2023-07-06 15:43:05 MISC: [INFO] 2023-07-06 15:43:05 VAL_FREQ: 1 [INFO] 2023-07-06 15:43:05 SAVE_FREQ: 1 [INFO] 2023-07-06 15:43:05 NUM_WORKERS: 0 [INFO] 2023-07-06 15:43:05 ==================Options================== [INFO] 2023-07-06 15:43:05 config=configs/faster_rcnn_mydata.yml [INFO] 2023-07-06 15:43:05 debug=False [INFO] 2023-07-06 15:43:05 device=cuda:0 [INFO] 2023-07-06 15:43:05 epochs=13 [INFO] 2023-07-06 15:43:05 gpu_id=0 [INFO] 2023-07-06 15:43:05 load=pretrained/13_Faster_RCNN.pt [INFO] 2023-07-06 15:43:05 local_rank=None [INFO] 2023-07-06 15:43:05 no_val=False [INFO] 2023-07-06 15:43:05 resume=False [INFO] 2023-07-06 15:43:05 save_path=None [INFO] 2023-07-06 15:43:05 tag=faster_rcnn_mydata [INFO] 2023-07-06 15:43:05 vis=True [INFO] 2023-07-06 15:43:05 =========================================== [INFO] 2023-07-06 15:43:05 =========================================== [INFO] 2023-07-06 15:43:05 val_trasforms: Compose([ ToTensorV2(always_apply=True, p=1.0, transpose_mask=False), ], p=1.0, bbox_params={'format': 'pascal_voc', 'label_fields': ['labels'], 'min_area': 0, 'min_visibility': 0, 'check_each_transform': True}, keypoint_params=None, additional_targets={}) [INFO] 2023-07-06 15:43:05 =========================================== Traceback (most recent call last):...........] ETA: 33m File "eval.py", line 59, in evaluate(model, val_dataloader, which_epoch, writer, logger, 'val') File "eval.py", line 20, in evaluate loss = model.evaluate(dataloader, epoch, writer, logger, data_name) File "D:\easy_detection\easy_detection\network\Faster_RCNN\Model.py", line 183, in evaluate return self.eval_mAP(dataloader, epoch, writer, logger, data_name) File "D:\easy_detection\easy_detection\network\base_model.py", line 95, in eval_mAP img = tensor2im(image).copy() File "D:\Anaconda\Anaconda3\envs\python37\lib\site-packages\mscv\image\image_io.py", line 14, in tensor2im x = x.data.cpu().numpy()[0] AttributeError: 'list' object has no attribute 'data'