Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 519 Bytes

README.md

File metadata and controls

9 lines (6 loc) · 519 Bytes

Imagenet pretraining script and model

This folder contains the script and model definition to pretrain ERFNet's encoder in Imagenet Data.

The script is an adaptation from the code in Pytorch Imagenet example. Please make sure that you have Imagenet dataset split in train and val folders before launching the script. Refer to that repository for instructions about usage and main.py options. Basic command:

python main.py <imagenet_folder_path>