efficientdet-pytorch icon indicating copy to clipboard operation
efficientdet-pytorch copied to clipboard

这是一个efficientdet-pytorch的源码,可以用于训练自己的模型。

Results 50 efficientdet-pytorch issues
Sort by recently updated
recently updated
newest added

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...

dependencies

使用自己的数据集进行训练时出现了这个错误,请问应该怎么解决呢? ./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 ![Image](https://github.com/user-attachments/assets/6f07f8a1-f14e-4707-b2ae-7d1eb771928f)