提交 d7f73dd8 编写于 作者: H HydrogenSulfate

add relabel=True in ResNet50_ReID to avoid label excess

上级 5b0a47bc
......@@ -32,7 +32,7 @@ Arch:
Head:
name: "CircleMargin"
margin: 0.35
scale: 64
scale: 64
embedding_size: 512
class_num: 3000
......@@ -56,48 +56,74 @@ Optimizer:
name: Cosine
learning_rate: 0.04
regularizer:
name: 'L2'
name: "L2"
coeff: 0.0001
# data loader for train and eval
DataLoader:
Train:
dataset:
name: ImageNetDataset
image_root: "dataset/LogoDet-3K-crop/train/"
cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt"
relabel: True
transform_ops:
- DecodeImage:
to_rgb: True
channel_first: False
- ResizeImage:
size: 224
- RandFlipImage:
flip_code: 1
- AugMix:
prob: 0.5
- NormalizeImage:
scale: 0.00392157
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ""
- RandomErasing:
EPSILON: 0.5
sampler:
name: PKSampler
batch_size: 128
sample_per_id: 2
drop_last: True
loader:
num_workers: 6
use_shared_memory: True
Eval:
Query:
dataset:
name: ImageNetDataset
image_root: "dataset/LogoDet-3K-crop/train/"
cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt"
image_root: "dataset/LogoDet-3K-crop/val/"
cls_label_path: "dataset/LogoDet-3K-crop/query_list.txt"
transform_ops:
- DecodeImage:
to_rgb: True
channel_first: False
- ResizeImage:
size: 224
- RandFlipImage:
flip_code: 1
- AugMix:
prob: 0.5
- NormalizeImage:
scale: 0.00392157
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ''
- RandomErasing:
EPSILON: 0.5
sampler:
name: PKSampler
order: ""
sampler:
name: DistributedBatchSampler
batch_size: 128
sample_per_id: 2
drop_last: True
loader:
num_workers: 6
drop_last: False
shuffle: False
loader:
num_workers: 8
use_shared_memory: True
Eval:
Query:
Gallery:
dataset:
name: ImageNetDataset
image_root: "dataset/LogoDet-3K-crop/val/"
cls_label_path: "dataset/LogoDet-3K-crop/query_list.txt"
image_root: "dataset/LogoDet-3K-crop/train/"
cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt"
transform_ops:
- DecodeImage:
to_rgb: True
......@@ -108,44 +134,18 @@ DataLoader:
scale: 0.00392157
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ''
order: ""
sampler:
name: DistributedBatchSampler
batch_size: 128
drop_last: False
shuffle: False
loader:
num_workers: 8
use_shared_memory: True
Gallery:
dataset:
name: ImageNetDataset
image_root: "dataset/LogoDet-3K-crop/train/"
cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt"
transform_ops:
- DecodeImage:
to_rgb: True
channel_first: False
- ResizeImage:
size: 224
- NormalizeImage:
scale: 0.00392157
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ''
sampler:
name: DistributedBatchSampler
batch_size: 128
drop_last: False
shuffle: False
name: DistributedBatchSampler
batch_size: 128
drop_last: False
shuffle: False
loader:
num_workers: 8
use_shared_memory: True
num_workers: 8
use_shared_memory: True
Metric:
Eval:
- Recallk:
topk: [1, 5]
- mAP: {}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册