mobilenet-yolov4-pytorch icon indicating copy to clipboard operation
mobilenet-yolov4-pytorch copied to clipboard

这是一个mobilenet-yolov4的库,把yolov4主干网络修改成了mobilenet,修改了Panet的卷积组成,使参数量大幅度缩小。

Results 52 mobilenet-yolov4-pytorch issues
Sort by recently updated
recently updated
newest added

您好 在使用resnet(图左) 和mobilenetv2(图右)作为主干网络训练自己的数据集(3000imgs 只有火焰一类) 出现了大量的FP样本 请问这是为什么 是因为框设置不合理吗 还是NMS设置不合理 另外 为什么使用了mobilenet2之后map会下降这么多 谢谢 ![QQ截图20220317135755](https://user-images.githubusercontent.com/71703688/158746348-be820ace-0ff9-47ed-8088-92e975c39745.png)

大佬您好,在summary函数中查看网络结构时,一直会有错误,显示为未解析的引用 'summary'。 ![image](https://user-images.githubusercontent.com/81509046/157248167-1f55a9b9-432e-4a51-8a27-d52d5caa5e60.png)

大佬你好,我使用你提供的主干网络然后训练了自己的权值文件,运行get_map.py 提示会有4952张图片,这个正常吗? 我有修改了(训练集+验证集)与测试集的比例,然后再次运行voc_annovation.py.产生新的txt,再次运行get_map.py发现还是4952张图片,请问这是怎么回事?

![image](https://user-images.githubusercontent.com/77728649/155135520-b28d5a52-d1ef-4b90-97df-9be4ab4a6871.png) 大佬,我用你给的densenet权重训练出现了这个问题,这是为什么呢

我做了如下修改:1、重新聚类了12个anchor,2、额外增加了一层检测层。那么预训练权重是不是不管用了?我不载入预训练权重开始训练,显示GPU内存不足。我用的是2060S,8G显存。请问这是显卡不够还是代码修改的问题? RuntimeError: CUDA out of memory. Tried to allocate 86.00 MiB (GPU 0; 8.00 GiB total capacity; 5.75 GiB already allocated; 0 bytes free; 5.80 GiB reserved in total by...

期望可以通过增加inference的batch_size,来提高gpu利用率。

# pw nn.Conv2d(inp, hidden_dim, 1, 1, 0, bias=False), nn.BatchNorm2d(hidden_dim), h_swish() if use_hs else nn.ReLU(inplace=True), # dw nn.Conv2d(hidden_dim, hidden_dim, kernel_size, stride, (kernel_size - 1) // 2, groups=hidden_dim, bias=False), nn.BatchNorm2d(hidden_dim), # Squeeze-and-Excite...