High-Resolution-Remote-Sensing-Semantic-Segmentation-PyTorch
High-Resolution-Remote-Sensing-Semantic-Segmentation-PyTorch copied to clipboard
PyTorch实现高分遥感语义分割(地物分类)
RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. 代码中错误定位在trainer.training(epoch)和writer.add_scalar('train_loss',...
# from class_names import gid_classes In the **dataset.py** file, the line **4**: > ```python >from class_names import gid_classes >``` But the file **class_names.py** just have the **def fifteen_classes** and **def...
很多文件png、jpg、pth缺少,很难训练
Mixup()
self.mixup_transform = sync_transforms.Mixup() sync_transforms.py中没有Mixup()这个函数啊
Mixup
HI,there is no mixup!
原始图片大小:2448*2448 切片大小:512*512 batch-size:16 GPU为3090,显存:24G
加载断点path
resnet.load_state_dict((torch.load("/root/data/others/yaoganbisai/code_6_7/models/pretrained_model/resnet50-19c8e357.pth"))) if args.model == 'deeplabv3_version_1'中,设置了path之后,会报错。错误在上面。这个resnet里面的pth是怎么生成的。
请问怎么获得您在github上贴的那种预测图?有代码参考嘛,我在test.py文件中获得的预测图结果不对
self.mixup_transform = sync_transforms.Mixup()请问这个问题,是需要将Mixup()直接去掉嘛