Skip to content

kevinrodrigues13/News_Image_Classification

Repository files navigation

News_Image_Classification

Required Dependencies

Cuda
Python
Pytorch

Step to train model

python Train.py --cuda

Step to test model

python Pred.py --image_dir test --model model_best.pth.tar --output_csvpath result.csv --cuda

Step to create visualztions on predictions from model

python Resultvisualization.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors