simplest_FasterRcnn_pytorch

simplest_FasterRcnn_pytorch copied to clipboard

Hi, I have this error in runing train.py!

3

你好，我非常感兴趣您的这个项目，觉得很简洁；但是当我在运行train.py 的时候，出现了以下的错误，您能帮忙解决一下吗？谢谢！ Traceback (most recent call last): File "train.py", line 25, in losses = model.train_step(img,bbox,label,scale,epoch) File "/home/a/Bcw_data/simplest_FasterRcnn_pytorch/model/faster_rcnn_vgg16.py", line 178, in train_step features, roi, rpn_loc_loss, rpn_cls_loss = self.train_extracor_and_rpn(img, bbox, scale, retain_graph=True)...

BCWang93

RT，我看到你的代码中时不时的要把数据从numpy上切换到cuda.tensor上或者从cuda.tensor切换到numpy上面，这一点我在chenyuntc/simple-faster-rcnn-pytorch的代码中也有发现。请问你知道有关这方面的技巧吗？让代码运行更快的技巧。就是应该在什么时候从numpy->cuda，而什么时候从cuda->numpy。或者如果你知道相关论文博客链接的话也可以。

pangkun248

running on ubuntu18.04,cuda10+cudnn7.5.0

PeterBishop0

请问如何才能改变batchsize的大小呢？

i4yyds

TypeError: __init__() got an unexpected keyword argument 'pretrained'

2

dear,when i run the test.py, has the problem as follow: Connected to pydev debugger (build 182.4505.26) Traceback (most recent call last): File "/home/yjs/Downloads/pycharm/helpers/pydev/pydevd.py", line 1664, in main() File "/home/yjs/Downloads/pycharm/helpers/pydev/pydevd.py", line...

yuanjinsheng