吕文玉
吕文玉
默认还是一样的;但是现在的代码是支持根据具体需求调节的 https://github.com/lyuwenyu/RT-DETR/blob/14971513a5ef805edaf434af1a11403d2f1a9036/rtdetrv2_pytorch/configs/rtdetrv2/include/rtdetrv2_r50vd.yml#L62-L65
目前不支持旋转框检测;需要适配下
暂时还没有适配
Please check that your onnxruntime is using GPU. ``` # pip install onnxruntime-gpu import onnxruntime as ort print(ort.get_device()) ```
> ```python > start_time = time.time() > output = sess.run( > # output_names=['labels', 'boxes', 'scores'], > output_names=None, > input_feed={'images': im_data.data.numpy(), "orig_target_sizes": size.data.numpy()} > ) > end_time = time.time() > ```...
**包含**`后处理`,**不包含**`图像预处理`
https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/configs/rtdetr/include/dataloader.yml#L29-L35
1. 这个要看你想要可视化什么吧 2. 没有相关的代码
哈哈哈研究明白了 可以提PR贡献一下
在loss里 具体实现采用的`vfl`