Repo moved here.
- If using a
conda environment:
conda create -n cathsim python=3.9
conda activate cathsim- Install the environment:
git clone -b [email protected]:tudorjnu/cathsim.git
cd cathsim
pip install -e .A quick way to have the enviromnent run with gym is to make use of the make_env function. The function automatically configures the environment and wraps it in a gym compatible environment.
from cathim.utils import make_env
env = make_env(
flatten_obs=True,
time_limit=300,
normalize_obs=False,
frame_stack=1,
)
obs = env.reset()
for _ in range(1):
action = env.action_space.sample()
obs, rewards, dones, info = env.step(action)
print(obs, rewards, dones, info)@article{jianu2022cathsim,
title={CathSim: An Open-source Simulator for Autonomous Cannulation},
author={Jianu, Tudor and Huang, Baoru and Abdelaziz, Mohamed EMK and Vu, Minh Nhat and Fichera, Sebastiano and Lee, Chun-Yi and Berthet-Rayne, Pierre and Nguyen, Anh and others},
journal={arXiv preprint arXiv:2208.01455},
year={2022}
}