efficientdet-pytorch
efficientdet-pytorch copied to clipboard
这是一个efficientdet-pytorch的源码,可以用于训练自己的模型。
Bumps [scipy](https://github.com/scipy/scipy) from 1.2.1 to 1.10.0. Release notes Sourced from scipy's releases. SciPy 1.10.0 Release Notes SciPy 1.10.0 is the culmination of 6 months of hard work. It contains many...
使用自己的数据集进行训练时出现了这个错误,请问应该怎么解决呢? ./aten/src/ATen/native/cuda/IndexKernel.cu:91: operator(): block: [0,0,0], thread: [1,0,0] Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds"` failed. ../aten/src/ATen/native/cuda/IndexKernel.cu:91: operator(): block: [0,0,0], thread: [2,0,0] Assertion `index >=...
标签框筛选问题
您好,想问一下代码在加载标签时有根据宽高比筛选标签框的操作吗,我发现训练出来的模型对细长直的目标检测效果比较差,有没有可能是在训练阶段这类极限宽高比的标签被筛掉了
在github代码efficientdet-pytorch/nets/efficientdet_training.py L48-L50,"重合度小于0.4需要参与训练 ",是否应该是"重合度大于0.4需要参与训练 "?因为下面计算loss使用的是大于0.5的部分 这一点可能是导致下面那位同学没有训练出结果的原因
您好,在使用您的FCOS模型时,训练了3次,都是在同一参数,数据集下训练的,可是最终得到的评估精度是不一样的,这个是正常的嘛?
训练d1去检测织物瑕疵(图片大小1280x1080,标签大小50x900,有些更小),训练出来map都是0.04这样的
Traceback (most recent call last): File "predict.py", line 14, in efficientdet = Efficientdet() File "/home/wwl/code/efficientdet-pytorch-master/efficientdet.py", line 87, in __init__ self.generate() File "/home/wwl/code/efficientdet-pytorch-master/efficientdet.py", line 101, in generate self.net.load_state_dict(torch.load(self.model_path, map_location=device)) File "/home/wwl/anaconda3/envs/EffcientNet/lib/python3.8/site-packages/torch/serialization.py",...
you are great 