magnus-haw / arcjetCV

desktop application for edge tracking of models in arc jet videos
Other
2 stars 0 forks source link

Optimize pytorch 2d CNN model #16

Closed magnus-haw closed 6 months ago

magnus-haw commented 7 months ago

Find the best model architecture for the new dataset. Move to start training on PFE

alexandrequ commented 7 months ago

Currently the xception model detects well the sample of the edge but not the shock.

alexandrequ commented 6 months ago

The xception model with 25 epoch is providing the best results