乌日达
具有统一联合分布对齐的无监督域自适应
先决条件:
Python3
PyTorch == 0.4.1(具有合适的CUDA和CuDNN版本)
火炬视觉== 0.2.0
脾气暴躁的
tqdm
资料集:
您需要在“ ./data”中的每个“ .txt”中修改图像的路径。
训练:
跑步 :
python train.py --config ../config/dann.yml --dataset Office-31 --src_address ../data/amazon.txt --tgt_address ../data/dslr.txt --src_test_address ../data/amazon.txt
引文:
如果您使用此代码进行研究,请考虑引用:
接触
如果您对我们的代码有任何,请随时联系 。
2023-03-01 20:14:51
42KB
Python
1