polish some issue: 1. install script; 2. fox; 3. remove useless dep; 4. imp dataset error msg. 5 better setup

Author: cjld

README.md

@@ -32,12 +32,9 @@ JNeRF is a benchmark toolkit and can be updated frequently, so installing in edi
 Thus any modifications made to JNeRF will take effect without reinstallation.
 
 ```shell
-sudo apt-get install tcl-dev tk-dev python3-tk
 git clone https://github.com/Jittor/JNeRF
 cd JNeRF
-python3.x -m pip install -r requirements.txt
-cd python
-python3.x -m pip install -e .
+python3.x -m pip install --user -e .
 ```
 If you have any installation problems for Jittor, please refer to [Jittor](https://github.com/Jittor/jittor)
 
@@ -97,6 +94,19 @@ python tools/extract_mesh.py --config-file ./projects/ngp/configs/ngp_base.py --
 ### Running other models
 
 Instructions of how to running other models are put in the contrib folder.
+```
+contrib/
+|---mipnerf/
+|---pixelnerf/
+|---plenoxel/
+```
+
+### Running fox dataset
+
+run builtin fox dataset:
+```
+python3.x ./tools/run_net.py --config ./projects/ngp/configs/ngp_fox.py
+```
 
 ## Performance
 

python/jnerf/__init__.py

@@ -6,5 +6,6 @@
 from . import utils
 
 # version must use ' instead of "
-__version__ = '0.1.0.0'
-
+__version__ = '0.1.1.0'
+from jittor_utils import LOG
+LOG.i("JNeRF version ", __version__)

python/jnerf/dataset/dataset.py

@@ -84,6 +84,7 @@ def load_data(self,root_dir=None):
             else:
                 json_data['frames']+=data['frames']
 
+        assert json_data is not None, f"dataset is not found at {root_dir}"
         ## init set  scale & offset
         if 'h' in json_data:
             self.H=int(json_data['h'])

python/jnerf/dataset/neus_dataset.py

@@ -4,7 +4,6 @@
 import numpy as np
 import os
 from glob import glob
-from icecream import ic
 from scipy.spatial.transform import Rotation as Rot
 from scipy.spatial.transform import Slerp
 

python/jnerf/models/samplers/neus_render/renderer.py

@@ -4,7 +4,6 @@
 import numpy as np
 import logging
 import mcubes
-from icecream import ic
 
 from jnerf.utils.config import get_cfg
 from jnerf.utils.registry import SAMPLERS

python/jnerf/runner/neus_runner.py

@@ -7,9 +7,7 @@
 import trimesh
 
 from shutil import copyfile
-from icecream import ic
 from tqdm import tqdm
-from pyhocon import ConfigFactory
 from jnerf.dataset.neus_dataset import NeuSDataset
 from jnerf.models.networks.neus_network import NeuS
 from jnerf.models.samplers.neus_render.renderer import NeuSRenderer

python/setup.py

@@ -0,0 +1,34 @@
+from setuptools import find_packages, setup
+import os
+import sys
+
+path = os.path.dirname(__file__)
+os.chdir(path)
+with open("./python/jnerf/__init__.py", "r") as f:
+    src = f.read()
+version = src.split("__version__")[1].split("'")[1]
+print("setup jnerf version", version)
+
+setup(
+    name="jnerf",
+    version=version,
+    description="NeRF benchmark based on Jittor",
+    author="jnerf",
+    url="https://github.com/Jittor/JNeRF",
+    packages=["jnerf"],
+    package_dir={'': 'python'},
+    package_data={'': [ "*"+"/*"*i for i in range(20)]},
+    install_requires=[
+        "jittor>=1.3.5.25",
+        "numpy",
+        "tqdm",
+        "opencv-python",
+        "Pillow",
+        "imageio",
+        "pyyaml",
+        "PyMCubes",
+        "trimesh",
+        "open3d"
+    ]
+)
+