Skip to content

Commit 6baa6a8

Browse files
baorepoPillar1989
authored andcommitted
configs: Create a new rtmdet folder, starting with nano size
1 parent 2a6e727 commit 6baa6a8

7 files changed

+9
-852
lines changed

configs/rtmdet_nano_8xb32_300e_coco_relu.py renamed to configs/rtmdet/rtmdet_nano_8xb32_300e_coco_relu.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
from mmengine.config import read_base
22

33
with read_base():
4-
from ._base_.default_runtime import *
5-
from ._base_.schedules.schedule_1x import *
6-
from .datasets.coco_detection import *
4+
from .._base_.default_runtime import *
5+
from .._base_.schedules.schedule_1x import *
6+
from ..datasets.coco_detection import *
77

88
from torchvision.ops import nms
99
from torch.nn import ReLU, BatchNorm2d
@@ -45,7 +45,7 @@
4545
from sscma.quantizer import RtmdetQuantModel
4646

4747
default_hooks.visualization = dict(
48-
type=DetVisualizationHook, draw=True, test_out_dir="works"
48+
type=DetVisualizationHook, draw=False, test_out_dir="works"
4949
)
5050

5151
visualizer = dict(type=DetLocalVisualizer, vis_backends=vis_backends, name="visualizer")
@@ -73,8 +73,8 @@
7373
type=RTMDet,
7474
data_preprocessor=dict(
7575
type=DetDataPreprocessor,
76-
mean=[0, 0, 0],
77-
std=[255, 255, 255],
76+
mean=[103.53, 116.28, 123.675],
77+
std=[57.375, 57.12, 58.395],
7878
bgr_to_rgb=False,
7979
batch_augments=None,
8080
),
@@ -160,7 +160,7 @@
160160
bbox_head=model["bbox_head"],
161161
data_preprocessor=model["data_preprocessor"], # data_preprocessor,
162162
)
163-
imdecode_backend = "torch"
163+
imdecode_backend = "pillow"
164164
train_pipeline = [
165165
dict(
166166
type=LoadImageFromFile,
@@ -219,7 +219,7 @@
219219
]
220220

221221
test_pipeline = [
222-
dict(type=LoadImageFromFile,imdecode_backend=imdecode_backend, backend_args=backend_args),
222+
dict(type=LoadImageFromFile,imdecode_backend=imdecode_backend, backend_args=None),
223223
dict(type=LoadAnnotations,imdecode_backend=imdecode_backend, with_bbox=True),
224224
dict(type=Resize, scale=imgsz, keep_ratio=True),
225225
dict(type=Pad, size=imgsz, pad_val=dict(img=(114, 114, 114))),
@@ -254,7 +254,7 @@
254254

255255

256256
val_dataloader = dict(
257-
batch_size=16,
257+
batch_size=32,
258258
num_workers=8,
259259
persistent_workers=True,
260260
pin_memory=True,
File renamed without changes.

configs/rtmdet_l_8xb32_300e_coco.py

Lines changed: 0 additions & 284 deletions
This file was deleted.

configs/rtmdet_m_8xb32_300e_coco.py

Lines changed: 0 additions & 16 deletions
This file was deleted.

0 commit comments

Comments
 (0)