O2U-Net + some modifications
O2U-Net implementation source git: link
python main.py --network=resnet50 --noise_rate 0.6 --remove_rate 0.8
(test)
python main.py --network=resnet50 --noise_rate 0.6 --remove_rate 0.85 --n_epoch1 1 --n_epoch2 2 --n_epoch3 2
2022/05
- Remove Caffe2 thread-pool leak warning: link
- 10 epoch experiment result
epoch:10 lr:0.009100 train_loss: 3.1395677614974975 test_accuarcy:55.330000 noise_accuracy:0.840500 top 0.1 noise accuracy:0.982200
- model: ResNet50, ResNet101
- python main.py --network=resnet101 --transforms=true (network: resnet50 / resnet101)
2022/06
- first stage, second stage: main.py
- third stage: curriculum.py : apply curriculum learning with masked dataset
- mod save directory
- Usage: python main.py --network=resnet50 --noise_rate 0.6 --remove_rate 0.8
- Add readme flow.jpg image
2022/07
- transform_ad.py: add more transform operations
TranslateX, TranslateY, Flip, Rotate, Posterize, AutoContrast