AlibabaResearch / AlibabaResearch/efficientteacher

用标准yolov5s转换为et模型后进行半监督训练,在训练过程中所有指标变为0

Aperta
#64 8 commenti 0 reazioni 0 assegnatari Vedi su GitHub
Lingua principale
Python
Stelle
813
Fork
126
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

# 我先自己训练了一个标准的yolov5s模型,然后用convert_yolov5_to_efficient.py脚本转换为efficient-yolov5s.pt,再进行半监督训练,配置文件如下:

project: './runs/yolov5_ssod'
adam: False
epochs: 200
weights: './data/et/fire_smoke/efficient-yolov5s.pt'
prune_finetune: False
linear_lr: True
hyp:
lr0: 0.001
hsv_h: 0.015
hsv_s: 0.7
hsv_v: 0.4
lrf: 1.0
scale: 0.9
burn_epochs: 20
no_aug_epochs: 0
warmup_epochs: 3

Model:
depth_multiple: 0.33 # model depth multiple
width_multiple: 0.50 # layer channel multiple
Backbone:
name: 'YoloV5'
activation: 'SiLU'
Neck:
name: 'YoloV5'
in_channels: [256, 512, 1024]
out_channels: [256, 512, 1024]
activation: 'SiLU'
Head:
name: 'YoloV5'
activation: 'SiLU'
anchors: [[10,13, 16,30, 33,23],[30,61, 62,45, 59,119],[116,90, 156,198, 373,326]] # P5/32]
Loss:
type: 'ComputeLoss'
cls: 0.3
obj: 0.7
anchor_t: 4.0

Dataset:
data_name: 'coco'
train: ../od_data/fire_smoke_filter/train.txt # 118287 images
val: ../od_data/fire_smoke_filter/test.txt # 5000 images
test: ../od_data/fire_smoke_filter/test.txt # 20288 of 40670 images, submit to https://competitions.codalab.org/competitions/20794^
target: ../od_data/fire_smoke_filter/unlabel.txt
nc: 2 # number of classes
np: 0 #number of keypoints
names: ['fire', 'smoke']
img_size: 640
batch_size: 8

SSOD:
train_domain: True
nms_conf_thres: 0.1
nms_iou_thres: 0.65
teacher_loss_weight: 3.0
cls_loss_weight: 0.3
box_loss_weight: 0.05
obj_loss_weight: 0.7
loss_type: 'ComputeStudentMatchLoss'
ignore_thres_low: 0.1
ignore_thres_high: 0.6
uncertain_aug: True
use_ota: False
multi_label: False
ignore_obj: False
pseudo_label_with_obj: True
pseudo_label_with_bbox: True
pseudo_label_with_cls: False
with_da_loss: False
da_loss_weights: 0.01
epoch_adaptor: True
resample_high_percent: 0.25
resample_low_percent: 0.99
ema_rate: 0.999
cosine_ema: True
imitate_teacher: False
ssod_hyp:
with_gt: False
mosaic: 1.0
cutout: 0.5
autoaugment: 0.5
scale: 0.8
degrees: 0.0
shear: 0.0
# 开始半监督训练后,经过几个epoch所有指标变为0,请问是什么问题呢?
![2023-04-11_16-07](https://user-images.githubusercontent.com/67748803/231097248-59ec59bf-453f-401f-b254-f55795658272.png)

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.