Skip to content

Commit 1d25c23

Browse files
committed
add requirements.
1 parent 07bf1b2 commit 1d25c23

File tree

3 files changed

+14
-1
lines changed

3 files changed

+14
-1
lines changed

.gitignore

-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
*.jpg
2-
*.txt
32

43
# compilation and distribution
54
__pycache__

README.md

+5
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@ This is a [MMDetection3D](https://github.com/open-mmlab/mmdetection3d) implement
1111

1212

1313
## Prerequisites
14+
The code is tested with Python3.7, PyTorch == 1.8, CUDA == 11.1, mmdet3d == 0.18.1, mmcv_full == 1.3.18 and mmdet == 2.14. We recommend you to use anaconda to make sure that all dependencies are in place. Note that different versions of the library may cause changes in results.
15+
1416
**Step 1.** Create a conda environment and activate it.
1517
```
1618
conda create --name demf python=3.7
@@ -41,5 +43,8 @@ We also provide pre-trained DeMF [here](https://drive.google.com/file/d/1s7mOJbz
4143
python -m torch.distributed.launch --nproc_per_node=8 --master_port=$PORT test.py --config configs/demf/demf_votenet.py --checkpoint $CHECKPOINT --eval mAP --launcher pytorch ${@:4}
4244
```
4345

46+
## Todo
47+
We will release the code of the DeMF (Fcaf3d based) soon.
48+
4449
## Citation
4550
If you find this work useful for your research, please cite our paper:

requirements.txt

+9
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
mmcv==1.6.0
2+
mmcv_full==1.3.18
3+
mmdet==2.14.0
4+
mmdet3d==0.18.1
5+
mmdet3d.egg==info
6+
mmseg==1.3.0
7+
mmsegmentation==0.14.1
8+
numpy==1.19.5
9+
torch==1.8.0+cu111

0 commit comments

Comments
 (0)