search

braun-steven / DAFNe

Code for our paper "DAFNe: A One-Stage Anchor-Free Deep Model for Oriented Object Detection".
MIT License
60 stars 12 forks source link

data_dir = os.environ["DAFNE_DATA_DIR"] #7

Closed ulkuacil64 closed 2 years ago

ulkuacil64 commented 2 years ago

What should be DAFNE_DATA_DIR ? Thank you.

braun-steven commented 2 years ago

It's the root directory of your datasets. If you're using the docker setting, the directory is defined here: https://github.com/steven-lang/DAFNe/blob/b13912041a263904cf26ca5f3468c6bc64ce800c/tools/run.py#L236

ulkuacil64 commented 2 years ago

Thank you

braun-steven commented 2 years ago

No worries. In case you're not using the docker setting, just prepend your command with the according environment variable:

NVIDIA_VISIBLE_DEVICES=0,1,2,3 DAFNE_DATA_DIR=<data-dir> ./tools/plain_train_net.py --num-gpus 4 --config-file ./configs/dota-1.0/1024.yaml

and replace <data-dir> with your desired directory.