coderhss / efficientdet-pytorch Goto Github PK

First of all, I am very impressed and grateful. It seems that you already ran the training process. It will be amazing if you share the checkpoint/pre-trained weight. @coderhss

你好，readme里面的网络结构图请问你是用什么工具可视化的呀？

i want to know ur pre-trained model's epoch.
and ur pre-trained model use all dataset in COCO?

Thank you for share your repository and implementation.
I'm training a model using my own dataset. The training process seems to work well, but when the model calls the evaluation function, I get this error:

File "/home/workstation/anaconda3/envs/efficientdet/lib/python3.7/site-packages/torch/nn/parallel/scatter_gather.py", line 55, in gather_map return Gather.apply(target_device, dim, *outputs) File "/home/workstation/anaconda3/envs/efficientdet/lib/python3.7/site-packages/torch/nn/parallel/_functions.py", line 54, in forward assert all(map(lambda i: i.is_cuda, inputs))

This error occurs in this call:

scores, labels, boxes = model(data['img'].permute(2, 0, 1).cuda().float().unsqueeze(dim=0))
Some idea?

This work has helped me a lot. There is no such good work on github.com across the entire network.