For training (default, cuda),
python train.py --train True --epochs 10 --record_interval 5
For testing, (default, cpu)
python train.py --load_path ckpt/model_ep9.pth
For mask to submission,
python mask_to_submission.py epoch epoch_number
-
Notifications
You must be signed in to change notification settings - Fork 0
One-punch24/UNet-for-Road-Seg
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published