GitHub

Install

Clone repo and install requirements.txt in a Python>=3.7.0 environment, including PyTorch>=1.7.

pip install -r requirements.txt  # install

Inference

python inf_ycb.py

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
__pycache__		__pycache__
assets		assets
data		data
datasets		datasets
experiments		experiments
lib		lib
models		models
replace_ycb_toolbox		replace_ycb_toolbox
tools		tools
trained_models/linemod		trained_models/linemod
utils		utils
vanilla_segmentation		vanilla_segmentation
.gitignore		.gitignore
CONTRIBUTING.md		CONTRIBUTING.md
Dockerfile		Dockerfile
LICENSE		LICENSE
README.md		README.md
__init__.py		__init__.py
best.pt		best.pt
detect.py		detect.py
download.sh		download.sh
export.py		export.py
hubconf.py		hubconf.py
inf_ycb.py		inf_ycb.py
inf_ycb_de.py		inf_ycb_de.py
inf_ycb_depth.py		inf_ycb_depth.py
requirements.txt		requirements.txt
seg_model.py		seg_model.py
setup.cfg		setup.cfg
train.py		train.py
tutorial.ipynb		tutorial.ipynb
val.py		val.py
yolo_det.py		yolo_det.py

Provide feedback