-
Setting this then running the RetinaNet guide should work
-
import torch
import torchvision
from torchviz import make_dot
model = torchvision.models.detection.retinanet_resnet50_fpn(pretrained=True)
model = model
model.eval()
x = [torch.rand(3, 300, 40…
-
**Describe the bug**
I trained my RetinaNet successful.
Now I tried to visualize the resulted bbox in the validation image.
My dataset is COCO, I know that it is pretrained with 80 classes, so the…
-
The README from DeepForest needs to be updated to represent any changes in the API.
Key steps
- [x] Copy docs from DeepForest
- [x] Look for changes in names, predict_generator -> predict_file…
bw4sz updated
3 years ago
-
### Describe the issue
We are trying to quantize our proprietary model based on RetinaNet using TensorRT's model optimization library. The following warning was raised: **"Please consider running pre…
-
-
My code worked well with GPU in Colab yesterday. But this morning it became very slow. So I suspect that CPU is used despite hardware accelerator is set to GPU in “change runtime type” explicitly. The…
bigmw updated
4 months ago
-
## Env
- GPU, RTX3090.
- OS, Ubuntu18.04
- Cuda 11.1
- TensorRT 7.2.3
Yolov5
在我尝试tensorrt加速yolov5时
我的步骤
1. 将自己训练的best.pt转成best.wts
2. 用sudo ./yolov5 -s yolov5s.wts yolov5s.engine s 命令生…
-
Broadcom has completed compliance tests for bert, bert-99, 3d-unet, retinanet, rnnt and stable-diffusion-xl. The compliance tests pass. We are having these two issues.
1. We are having trouble crea…
-
Hi,
I just wanted to know if it can be used as an object detection model too for detecting bounding Boxes etc for the documents.
I have [asked a similar question here](https://stackoverflow.com/q…