search

tfg-carvarsou / Vehicle-Classification

:red_car: Vehicle detection and classification project (2024)
Apache License 2.0
0 stars 0 forks source link

Re-entrenamiento de todos los modelos #14

Closed carvarsou closed 1 month ago

carvarsou commented 3 months ago

Una vez que la webapp esté terminada (frontend y backend), hay que hacer finetuning a todos los modelos para obtener mejores resultados en Google Colab.

Detección:

  1. Img size: 256 > 640
  2. Batch size: 8 > 16
  3. Num workers: 4 > 8
  4. Epochs: 5 > 20
  5. Optimizer: SGD
  6. Conf threshold: 0.3

Clasificación:

  1. Img size: 224
  2. Batch size: 16
  3. Num workers: 4 > 8
  4. Epochs: 5 > 20
  5. Optimizer: SGD
  6. Conf threshold: 0.3