TRACER: Texture-Robust Affordance Chain-of-Thought for Deformable-Object Refinement

Code is tested under Pytorch 1.12.1, python 3.7, and CUDA 11.3

pip install -r requirements.txt

Download the Fine-AGDDO15 dataset from Baidu Pan[D5FS].(you can annotate your own one-shot data in the same format).

Put the data in the dataset folder with the following structure:

dataset 
├── one-shot-seen
└── Seen

python train.py
python test.py --model_file <PATH_TO_MODEL>