{"id":13627671,"url":"https://github.com/TuSimple/DarkRank","last_synced_at":"2025-04-17T00:32:12.991Z","repository":{"id":68971699,"uuid":"130073392","full_name":"TuSimple/DarkRank","owner":"TuSimple","description":null,"archived":false,"fork":false,"pushed_at":"2018-04-18T14:24:02.000Z","size":596,"stargazers_count":54,"open_issues_count":2,"forks_count":8,"subscribers_count":3,"default_branch":"master","last_synced_at":"2025-04-06T02:51:15.513Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":null,"language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/TuSimple.png","metadata":{"files":{"readme":"readme.md","changelog":null,"contributing":null,"funding":null,"license":null,"code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null}},"created_at":"2018-04-18T14:17:07.000Z","updated_at":"2025-02-26T10:34:51.000Z","dependencies_parsed_at":"2024-01-14T08:06:29.370Z","dependency_job_id":null,"html_url":"https://github.com/TuSimple/DarkRank","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TuSimple%2FDarkRank","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TuSimple%2FDarkRank/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TuSimple%2FDarkRank/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TuSimple%2FDarkRank/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/TuSimple","download_url":"https://codeload.github.com/TuSimple/DarkRank/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":249293016,"owners_count":21245669,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":[],"created_at":"2024-08-01T22:00:37.038Z","updated_at":"2025-04-17T00:32:12.532Z","avatar_url":"https://github.com/TuSimple.png","language":"Python","readme":"#### Introduction\nThis is the repo for our AAAI-18 paper \"DarkRank: Accelerating Deep Metric Learning via Cross Sample Similarities\" \nIf you found this useful, please cite our paper.\n\n#### Preparing MXNet\n1. make a working directory\n```\nmkdir darkrank\n```\n\n2. clone the lastest stable mxnet\n```\ngit clone --recursive https://github.com/apache/incubator-mxnet.git mxnet-darkrank\n```\n\n3. clone LSoftmax from luoyetx\n```\ngit clone https://github.com/luoyetx/mx-lsoftmax.git lsoftmax\n```\n\n4. copy LSoftmax to mxnet\n```\ncp lsoftmax/operator/lsoftmax-inl.h lsoftmax/operator/lsoftmax.cc lsoftmax/operator/lsoftmax.cu mxnet-darkrank/src/operator/\n```\n\n5. install dependency and make mxnet\n```\nsudo apt-get update\nsudo apt-get install -y build-essential git\nsudo apt-get install -y libopenblas-dev liblapack-dev\nsudo apt-get install -y libopencv-dev\n```\n\n```\ncd mxnet-darkrank\nmake -j $(nproc) USE_OPENCV=1 USE_BLAS=openblas USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1\n```\n\n6. install mxnet\n```\npip install -e mxnet-darkrank/python \n```\n\n7. clone darkrank\n\n\n#### Preparing Data\n1. Download Market-1501-v15.09.15.zip from http://www.liangzheng.org/Project/project_reid.html\n2. unzip Market-1501-v15.09.15.zip into `data` directory\n3. run generate_rec.sh\n\n#### Training\n1. download imagenet pretrain\n```\nwget http://data.dmlc.ml/models/imagenet/inception-bn/Inception-BN-0126.params -O models/inception-bn-0126.params\nwget http://data.dmlc.ml/models/imagenet/nin/nin-0000.params -O models/nin-0000.params\n```\n2. teacher baseline\n```\npython reid.py --mode inception-cls-LMNN-Market1501 --even-iter --data-dir data/Market-list --num-examples 13314 --num-id 751 --lr 0.01 --num-epochs 100 --train-file train --student-network inception-bn --student-params-prefix inception-bn --student-params-epoch 126 --gpus 0\n```\n\n3. student baseline\n```\npython reid.py --mode nin-cls-LMNN-Market1501 --even-iter --data-dir data/Market-list --num-examples 13314 --num-id 751 --lr 0.01 --num-epochs 100 --train-file train --student-network nin-head-bn --student-params-prefix nin --student-params-epoch 0 --gpus 0\n```\n\n4. darkrank\n```\npython reid.py --mode distill-listnet-Market1501 --even-iter --data-dir data/Market-list --num-examples 13314 --num-id 751 --num-epochs 100 --train-file train --student-network nin-head-bn --student-params-prefix nin --student-params-epoch 0 --teacher-network inception-bn --teacher-params-prefix inception-cls-LMNN-Market1501-1524036667 --teacher-params-epoch 100 --lr 5e-4 --score-power 2.0 --embedding-l2-norm 3.5 --list-length 4 --loss-weight-listnet 16.0 --gpus 0\n```\n\n#### Testing\nextract 1024d `l2_norm_output` and do standard test\n","funding_links":[],"categories":["\u003ca name=\"Vision\"\u003e\u003c/a\u003e2. Vision"],"sub_categories":["2.6 ReID"],"project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2FTuSimple%2FDarkRank","html_url":"https://awesome.ecosyste.ms/projects/github.com%2FTuSimple%2FDarkRank","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2FTuSimple%2FDarkRank/lists"}