Fix/torchvision update

rtdetr_pytorch/README.md

@@ -107,3 +107,10 @@ python tools/export_onnx.py -c configs/rtdetr/rtdetr_r18vd_6x_coco.yml -r path/t
 
 2. add `-t path/to/checkpoint` (optinal) to tuning rtdetr based on pretrained checkpoint. see [training script details](./tools/README.md).
 </details>
+
+
+<details open>
+<summary>list of wheel release</summary>
+
+rtrdetr-0.1.1-py3-none-any.whl - updated to use torchvision =< 0.16.0
+

rtdetr_pytorch/configs/dataset/coco_detection.yml

@@ -6,9 +6,9 @@ remap_mscoco_category: True
 train_dataloader: 
   type: DataLoader
   dataset: 
-    type: CocoDetection
-    img_folder: ./dataset/coco/train2017/
-    ann_file: ./dataset/coco/annotations/instances_train2017.json
+    type: HRSC2016Dataset
+    img_folder: ../data/HRSC2016-MS/AllImages
+    ann_folder: ../data/HRSC2016-MS/Annotations
     transforms:
       type: Compose
       ops: ~
@@ -21,9 +21,9 @@ train_dataloader:
 val_dataloader:
   type: DataLoader
   dataset: 
-    type: CocoDetection
-    img_folder: ./dataset/coco/val2017/
-    ann_file: ./dataset/coco/annotations/instances_val2017.json
+    type: HRSC2016Dataset
+    img_folder: ../data/HRSC2016-MS/AllImages
+    ann_folder: ../data/HRSC2016-MS/Annotations
     transforms:
       type: Compose
       ops: ~ 

rtdetr_pytorch/configs/rtdetr/include/dataloader.yml

@@ -8,15 +8,15 @@ train_dataloader:
       ops:
         - {type: RandomPhotometricDistort, p: 0.5}
         - {type: RandomZoomOut, fill: 0}
-        - {type: RandomIoUCrop, p: 0.8}
-        - {type: SanitizeBoundingBox, min_size: 1}
+        - {type: RandomIoUCrop}
+        - {type: SanitizeBoundingBoxes, min_size: 1}
         - {type: RandomHorizontalFlip}
         - {type: Resize, size: [640, 640], }
         # - {type: Resize, size: 639, max_size: 640}
         # - {type: PadToSize, spatial_size: 640}
-        - {type: ToImageTensor}
-        - {type: ConvertDtype}
-        - {type: SanitizeBoundingBox, min_size: 1}
+        - {type: ToImage}
+        - {type: ToDtype}
+        - {type: SanitizeBoundingBoxes, min_size: 1}
         - {type: ConvertBox, out_fmt: 'cxcywh', normalize: True}
   shuffle: True
   batch_size: 4
@@ -31,8 +31,8 @@ val_dataloader:
         # - {type: Resize, size: 639, max_size: 640}
         # - {type: PadToSize, spatial_size: 640}
         - {type: Resize, size: [640, 640]}
-        - {type: ToImageTensor}
-        - {type: ConvertDtype}
+        - {type: ToImage}
+        - {type: ToDtype}
   shuffle: False
   batch_size: 8
   num_workers: 4

rtdetr_pytorch/configs/rtdetr/include/rtdetr_r50vd.yml

@@ -65,8 +65,8 @@ RTDETRPostProcessor:
 
 
 SetCriterion:
-  weight_dict: {loss_vfl: 1, loss_bbox: 5, loss_giou: 2,}
-  losses: ['vfl', 'boxes', ]
+  weight_dict: {loss_vfl: 1, loss_bbox: 5, loss_giou: 2, loss_angles: 3}
+  losses: ['vfl', 'boxes', 'angles']
   alpha: 0.75
   gamma: 2.0
 

rtdetr_pytorch/requirements.txt

@@ -1,5 +1,5 @@
 torch==2.0.1
-torchvision==0.15.2
+torchvision==0.16.0
 onnx==1.14.0
 onnxruntime==1.15.1
 pycocotools

rtdetr_pytorch/setup.py

@@ -0,0 +1,19 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='rtrdetr',
+    version='0.2a5',
+    author='Your Name',
+    author_email='your.email@example.com',
+    description='A short description of the package',
+    long_description=open('README.md').read(),
+    long_description_content_type='text/markdown',
+    url='https://example.com/your-package',
+    packages=find_packages(),
+    classifiers=[
+        'Programming Language :: Python :: 3',
+        'License :: OSI Approved :: MIT License',
+        'Operating System :: OS Independent',
+    ],
+    python_requires='>=3.6',
+)

rtdetr_pytorch/src/__init__.py

@@ -1,5 +1,4 @@
-
-from . import data 
+from . import data
 from . import nn
 from . import optim
-from . import zoo
+from . import zoo