A91A981E的博客

写在前面最近需要搞一个目标检测加上特征提取的小项目，找方法的时候看到了CenterNet这个网络模型，论文里的图表表现出CenterNet性能完虐YOLOv3，如下图看着还是非常诱惑的，所以试试用这个。论文里也给了官方代码，但是还是有不少问题。本机环境从整个配置过程来看，主要问题集中在DCNv2这个网络的编译上。经过不少的尝试，最终在我的环境下编译成功：PyTorch 1.5CUDA 10.2g++ 7NVIDIA Driver 470.57.02Ubuntu 20.04Pytho

写在前面最近使用图神经网络写了一些东西，觉得图神经网络还是挺有意思的，所以撰写一下学习记录，免得后面自己忘了。主要参考的内容放在最后了。由于本人小菜鸡一个，所以本文中尽量避免公式以及数学相关的内容背景自从AlexNet掀起深度学习浪潮之后，卷积神经网络(Convolution Neural Network, CNN)、深度神经网络(Deep Neural Network, DNN)、循环神经网络(Recurrent Neural Network, RNN)等网络结构均在计算机视觉、自然语言处理等领域大

问题描述在训练神经网络的时候，在... loss_sum = torch.sum(...) self.optimizer.zero_grad() -->loss_sum.mean().backward() self.optimizer.step()...报错RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn尝试解决情况一这是由于PyTorch在