Поезд Дар knet на 2 классах, но МАР снижается? - PullRequest
0 голосов
/ 18 марта 2020

Я пытаюсь потренировать дар knet с помощью вилки AlexeyAB на двух наборах изображений (лицо и руки), оба довольно большие> 2000 изображений, в среднем с разрешением 1000x600. Но после некоторой тренировки MAP резко падает. Похоже, что один из классов "обучен": mAPdown.jpg

Настроен как управляемый: классы = 2, фильтры = 21, cfg:

[net]
# Testing
# batch=1
# subdivisions=1
# Training
batch=64
subdivisions=16
width=512
height=512
channels=3
momentum=0.9
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1

learning_rate=0.001
burn_in=1000
max_batches = 10000
policy=steps
steps=8000,9000
scales=.1,.1

В результате (основываясь на некоторых вариациях, я пробовал 416 как размер и разные предварительно обученные веса) либо исчезают руки: nohands.jpg или лица: noface.jpg

Любые идеи приветствуются!

...