mobilenet-yolov4-pytorch

mobilenet-yolov4-pytorch copied to clipboard

老师，我电脑cpu训练，然后运行train.py时候训练的很慢，python3.6的，torch1.2的，torch vision0.4的，训练模型一天一夜都没运行完事，然后我运行其他的predict.py显示cuda错误， 1、是没有训练出来结果就不能进行下一步是吗，我电脑很慢，能不能给你训练好的模型发我一下，我跟你环境差不多，代码也没改，就改了你说的那部分， 2、我用的cpu，怎么会有cuda报错呢 3、import onnxsim怎么安装，还是不耽误？ 我是看了你的b站视频，修改了voc_annotation.py里面的annotation_mode=2，生成2007_train.txt和2007_val.txt。然后train.py训练一天一夜我就暂停了。之后yolo.py里面model_path以及classes_path， "model_path" : 'D:/opencv-4.7.0-windows/shenjieke/logs/best_epoch_weights.pth', "classes_path" : 'D:/opencv-4.7.0-windows/shenjieke/model_data/new_classes.txt', best_epoch_weights.pth是我自己瞎放的，因为模型没训练出来 在logs文件夹里目前有best_epoch_weights.pth和last_epoch_weights.pth 目前运行predict.py也没有框。 归根结底是不是就是模型没有训练出来所以下一步不好使？那我也没用cuda呀，谢谢up主。

zgeh994

mobilenetv1权重的由来？

1

您好！想问一下百度云里面mobilenetv1_weight.pth权重是来自于哪里？是不是基于imagenet的预训练模型？能否提供一下原先的该模型的imagenet的训练地址？拜托了。我一直在pytorch或者torchvision代码中找不到mobilenetv1的预训练pth文件，只有v2v3的，很奇怪。

myzhuang

你好，有没有cfg文件

1

lanshihhh

训练自己数据集预测为空

3

博主你好，我在使用yolov4+mobilenet和resnet50时使用自己数据集训练，训练过程map有值很小，但是预测图片无法得到结果。在使用VOC数据集时训练一次即可得到较好结果。训练过程之前参考过yolov3可以得到正常跑并预测图片，可以确定该修改的都改了，请问如何改正。

YuZhouShui

预测问题

1

在每十轮进行map计算时，仅有map值，其余值均为0，训练好进行预测时并不显示标框，但运行get_map.py时却能输出评价指标结果 65.29% = crack AP || score_threhold=0.5 : F1=0.00 ; Recall=0.00% ; Precision=0.00% mAP = 65.29%。 请问这是什么问题呢？谢谢

yiseni

b导你好，一直在CSDN上跟着您学习，最近遇到了一个问题，想请教一下您， 就是我把YOLOv4主干网络替换成mobilentv2，使用yolov4_mobilenet_v2_voc.pth权重训练，训练出来的精度只有一点几，F1直接为0。 Calculate Map. 1.37% = person AP || score_threhold=0.5 : F1=0.00 ; Recall=0.07% ; Precision=100.00% mAP = 1.37% Get map done. 这是什么问题啊？我看了您的问题汇总，该改的地方我都改了，我试了一下predict.py 图片直接是黑的 什么都没有

lyl-9

导儿，可以在你的yolo v5中直接加入mobilenet吗

4

Vision-LYF

运行summary时出现了报错

2

import torch from thop import clever_format, profile from torchsummary import summary from nets.yolo import YoloBody if __name__ == "__main__": input_shape = [416, 416] anchors_mask = [[6, 7, 8], [3, 4,...

Vision-LYF

我一块3080 12G显卡，复现不了你的精度正常吗，在voc数据集上平均每个模型掉5个AP值。是因为我的显存较小，无法大batch训练导致的吗

2

WDQGO

训练VOC数据集时的参数以及电脑配置

3

B导，我在你mobilenet-yolov4-pytorch的基础上换了一个ghostnetv2网络。现在我想和你之前的主干ghostnetv1、mobilenetv1-3等网络对比一下性能，但是不知道您之前使用VOC数据集训练网络时给的参数配置，想麻烦B导给个epoch，batch-size，optimizer，等其它参数呢。

Jie-Huangi