Skip to content

Releases: bobvo23/DANN-Plant-Counting

DANN for MNIST -> MNISTM

24 Feb 04:03
Compare
Choose a tag to compare

Bug fix: now default to train on GPU, fall back to CPU if needed.

Fully working adaptation model based on DANN architecture
Source dataset: MNIST
Target dataset: MNISTM

MNIST ACC: 99%
MNISTM ACC: 93.5% | Without Adaptation: 51%

Train:
python3 train.py -c config.json
Test:
python3 test.py -c config.json -r [checkpoint]

DANN for MNIST -> MNISTM

24 Feb 00:43
Compare
Choose a tag to compare

Fully working adaptation model based on DANN architecture
Source dataset: MNIST
Target dataset: MNISTM

MNIST ACC: 99%
MNISTM ACC: 93.5% | Without Adaptation: 51%

Train:
python3 train.py -c config.json
Test:
python3 test.py -c config.json -r [checkpoint]