diff --git a/README.md b/README.md index 0e23518..251fc58 100644 --- a/README.md +++ b/README.md @@ -8,9 +8,20 @@ A pytorch implementation of the "Selective Convolutional Descriptor Aggregation" [largestConnectComponent](https://blog.csdn.net/xuyangcao123/article/details/81023732) ## Details +install requirements + + +``` + pip install -r requirements.txt; +``` + On CUB and split dataset in 'CUB_200.py'. random split CUB-200-2011 results: ||top1|top5 ---|:--:|---: CUB|-|- + + + +>Wei X S , Luo J H , Wu J , et al. Selective Convolutional Descriptor Aggregation for Fine-Grained Image Retrieval[J]. IEEE Transactions on Image Processing, 2017, 26(6):2868-2881. \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..0a1d203 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,7 @@ +scipy==1.3.1 +scikit_image==0.16.2 +torchvision==0.4.2 +numpy==1.17.3 +torch==1.3.1 +Pillow==7.1.2 +skimage==0.0