efficientdet-pytorch
efficientdet-pytorch copied to clipboard
rwightman
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
Found this commit, but did not check it. Looks like as bug fix: https://github.com/ceshine/efficientdet-pytorch/commit/b8ccb5247533645db523d17a8dec4ae9ad612fe9
Hi! Thanks for your repo, it's very useful. But i met with some problem when tried to create model from pretrained checkpoint. I'm using effdet==0.3.0 package. When i run the...
PyTorch 1.12 ``` RuntimeError: The following operation failed in the TorchScript interpreter. Traceback of TorchScript (most recent call last): File "/xxxxx/efficientdet-pytorch/effdet/loss.py", line 210, in loss_fn cls_losses.append(cls_loss.sum()) # FIXME reference code...
Hi! I'm trying to train **tf_efficientdet_d3** with your code on custom coco-like data. I'm training on Google Colab. ` !python3 train.py '/content/drive/MyDrive/dataset' --model tf_efficientdet_d3 --num-classes 60 --pretrained -b 1 --save-images...
Hi Ross, thanks for sharing this repo! Curious if you have any plans on releasing a model trained on `OpenImages`? **Is your feature request related to a problem? Please describe.**...
The paper lists an easy way to use the model for segmentation. I really hope there is enough flexibility in your code to allow for that alteration. `Following [16], we...
I am trying to get the model trained on a custom dataset using PyTorch framework exported to ONNX and to further convert to TensorRT and run on jetson nano. However,...
As per your given TODOs in the README, I thought it would be nice to add initial mosaic augmentation to this codebase. Original implementation is from [YoloV5 repo](https://github.com/ultralytics/yolov5/), which I...
Hi Master. Can you add code to visualize the detection effect? I can't find the code for visual inspection in this repository.
**Is your feature request related to a problem? Please describe.** I am using custom augmentation to train the effdet. And I don`t know good way to integrate my augmentation code...