Saya melatih YOLOv3 melalui kerangka Darknet. Setiap 1000 iterasi itu menghemat bobot tetapi pada akhirnya, Darknet mengevaluasi semua bobot, dan menggunakan yang terbaik. Mereka disimpan dalam file terpisah "yolov3_best.weights".

Saya ingin mencari tahu, iterasi mana yang digunakan untuk file ini. Saya mencoba sejauh ini:

  • gunakan bobot dalam tes pengenalan melalui terminal dan periksa hasilnya
  • membuka file best.weights melalui Editor dan mencarinya

Tapi aku tidak bisa menemukannya.

Apakah ada yang punya solusi?

Terima kasih di muka.

0
mirArnold 22 April 2020, 22:11

1 menjawab

Jawaban Terbaik

Jadi, tidak jelas bagaimana cara mengetahui langkah/Epoch iterasi mana yang digunakan untuk file best_weights, jadi saya melakukan hal berikut:

Saya menulis skrip, yang menggunakan file "yolov3_best.weights" untuk mendeteksi semua kelas dalam testset, membandingkannya dengan label-data saya dan menghitung ingatan metrik, presisi, dan f1. Saya melakukan ini juga untuk semua bobot lainnya, yang darknet simpan secara default (setiap 1000 langkah iterasi) dan membandingkan hasilnya.

Pada akhirnya saya menemukan, bahwa "yolov3_best.weights" bukan yang terbaik untuk metrik saya, jadi saya memilih yang memiliki nilai ingatan tertinggi (tetapi yang lain dapat memilih sesuai dengan metrik yang harus dioptimalkan untuk kasus yang diperlukan ).

Semoga ini bisa membantu orang lain.

1
MirTh 26 April 2020, 14:12