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

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

Results 89 ssd-pytorch issues
Sort by recently updated
recently updated
newest added

错误提示: Traceback (most recent call last): File "D:/python_code/ssd网络/ssd-pytorch-master/train.py", line 111, in images = batch[0] TypeError: 'ExceptionWrapper' object does not support indexing Process finished with exit code 1 附batch代码 for iteration,...

您好,我在仿写get_dr_txt.py过程中发现检测的结果中有的图像未检测到任何目标,全是背景,所以并不会生成对应的txt,那么get_map.py也会报错,请问您遇到过这个问题么?

如果图片本身就是黑白图的话,就不用减掉均值进行训练了。

您好,看了您的哔哩哔哩视频讲解之后有一个问题向请教一下。 'feature_maps': [38, 19, 10, 5, 3, 1], 'steps': [8, 16, 32, 64, 100, 300], 就是这个steps的计算过程,按照您的视频讲解是300/8≈38,前两个数(8,16)是符合的,但是第三个数32不符合啊,如果改成30,那么300/30=10了?为什么是32而不是30?

Traceback (most recent call last): File "C:\ssd-pytorch-master\get_dr_txt.py", line 77, in ssd.detect_image(image_id,image) File "C:\ssd-pytorch-master\get_dr_txt.py", line 25, in detect_image preds = self.net(photo) File "C:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 532, in __call__ result = self.forward(*input, **kwargs)...

建议不要将数据集放入仓库,不然容量太大,说实话下载有难度。谢谢

在ssd.py中第56行 model.load_state_dict(torch.load(self.model_path)) 中的self.model_path直接改为log文件夹下训练好的自己的数据集的权重文件