Skip to content

Commit d7f73dd

Browse files
add relabel=True in ResNet50_ReID to avoid label excess
1 parent 5b0a47b commit d7f73dd

File tree

1 file changed

+55
-55
lines changed

1 file changed

+55
-55
lines changed

ppcls/configs/Logo/ResNet50_ReID.yaml

+55-55
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ Arch:
3232
Head:
3333
name: "CircleMargin"
3434
margin: 0.35
35-
scale: 64
35+
scale: 64
3636
embedding_size: 512
3737
class_num: 3000
3838

@@ -56,48 +56,74 @@ Optimizer:
5656
name: Cosine
5757
learning_rate: 0.04
5858
regularizer:
59-
name: 'L2'
59+
name: "L2"
6060
coeff: 0.0001
6161

6262
# data loader for train and eval
6363
DataLoader:
6464
Train:
6565
dataset:
66+
name: ImageNetDataset
67+
image_root: "dataset/LogoDet-3K-crop/train/"
68+
cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt"
69+
relabel: True
70+
transform_ops:
71+
- DecodeImage:
72+
to_rgb: True
73+
channel_first: False
74+
- ResizeImage:
75+
size: 224
76+
- RandFlipImage:
77+
flip_code: 1
78+
- AugMix:
79+
prob: 0.5
80+
- NormalizeImage:
81+
scale: 0.00392157
82+
mean: [0.485, 0.456, 0.406]
83+
std: [0.229, 0.224, 0.225]
84+
order: ""
85+
- RandomErasing:
86+
EPSILON: 0.5
87+
sampler:
88+
name: PKSampler
89+
batch_size: 128
90+
sample_per_id: 2
91+
drop_last: True
92+
93+
loader:
94+
num_workers: 6
95+
use_shared_memory: True
96+
Eval:
97+
Query:
98+
dataset:
6699
name: ImageNetDataset
67-
image_root: "dataset/LogoDet-3K-crop/train/"
68-
cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt"
100+
image_root: "dataset/LogoDet-3K-crop/val/"
101+
cls_label_path: "dataset/LogoDet-3K-crop/query_list.txt"
69102
transform_ops:
70103
- DecodeImage:
71104
to_rgb: True
72105
channel_first: False
73106
- ResizeImage:
74107
size: 224
75-
- RandFlipImage:
76-
flip_code: 1
77-
- AugMix:
78-
prob: 0.5
79108
- NormalizeImage:
80109
scale: 0.00392157
81110
mean: [0.485, 0.456, 0.406]
82111
std: [0.229, 0.224, 0.225]
83-
order: ''
84-
- RandomErasing:
85-
EPSILON: 0.5
86-
sampler:
87-
name: PKSampler
112+
order: ""
113+
sampler:
114+
name: DistributedBatchSampler
88115
batch_size: 128
89-
sample_per_id: 2
90-
drop_last: True
91-
92-
loader:
93-
num_workers: 6
116+
drop_last: False
117+
shuffle: False
118+
loader:
119+
num_workers: 8
94120
use_shared_memory: True
95-
Eval:
96-
Query:
121+
122+
Gallery:
97123
dataset:
98124
name: ImageNetDataset
99-
image_root: "dataset/LogoDet-3K-crop/val/"
100-
cls_label_path: "dataset/LogoDet-3K-crop/query_list.txt"
125+
image_root: "dataset/LogoDet-3K-crop/train/"
126+
cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt"
101127
transform_ops:
102128
- DecodeImage:
103129
to_rgb: True
@@ -108,44 +134,18 @@ DataLoader:
108134
scale: 0.00392157
109135
mean: [0.485, 0.456, 0.406]
110136
std: [0.229, 0.224, 0.225]
111-
order: ''
137+
order: ""
112138
sampler:
113-
name: DistributedBatchSampler
114-
batch_size: 128
115-
drop_last: False
116-
shuffle: False
117-
loader:
118-
num_workers: 8
119-
use_shared_memory: True
120-
121-
Gallery:
122-
dataset:
123-
name: ImageNetDataset
124-
image_root: "dataset/LogoDet-3K-crop/train/"
125-
cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt"
126-
transform_ops:
127-
- DecodeImage:
128-
to_rgb: True
129-
channel_first: False
130-
- ResizeImage:
131-
size: 224
132-
- NormalizeImage:
133-
scale: 0.00392157
134-
mean: [0.485, 0.456, 0.406]
135-
std: [0.229, 0.224, 0.225]
136-
order: ''
137-
sampler:
138-
name: DistributedBatchSampler
139-
batch_size: 128
140-
drop_last: False
141-
shuffle: False
139+
name: DistributedBatchSampler
140+
batch_size: 128
141+
drop_last: False
142+
shuffle: False
142143
loader:
143-
num_workers: 8
144-
use_shared_memory: True
144+
num_workers: 8
145+
use_shared_memory: True
145146

146147
Metric:
147148
Eval:
148149
- Recallk:
149150
topk: [1, 5]
150151
- mAP: {}
151-

0 commit comments

Comments
 (0)