- 博客(4)
- 收藏
- 关注
原创 训练SSD时报错IndexError: Too many indices for array:Array is 1-dimensional,but 2 were indexed
训练SSD网络的时候,出现报错:i。
2023-11-02 12:48:14
449
1
原创 一招解决本科毕设,使用自己的数据集训练网络(ResNet分类、YOLO目标检测、Unet、SwinTransformer图像分割)
ResNet:基于图像进行分类,可以替换为任意输入图像,给定不同的类比可以实现分类,基础网络为实现医学图像疾病分类,基础分类为2分类,多分类需要调整num_class大小。Unet:可以实现图像的分割,需要label,可以使用自己给定的任意数据集实现对应图像分割的结果。任务三:基于Yolov5、YOLOv6的目标检测(可输入视频,可做对比提升改进)任务一:基于ResNet的图像分类。任务二:基于Unet的图像分割。
2023-10-31 21:53:22
745
2
原创 使用自己的数据集训练SwinTransformer时交叉熵出现:RuntimeError: CUDA error: an illegal memory access was encountered
解决!!!RuntimeError: CUDA error: an illegal memory access was encountered
2023-10-31 20:32:18
289
1
基于Swin-Trasnformer的图像分割、语义分割
2023-10-31
YoLoV6目标检测代码
2023-10-31
基于YoloV5的目标检测,可用于毕业设计
2023-10-31
基于ResNet的图像分类和基于Unet的图像分割
2023-10-31
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人