coordinate_prediction

data preparation

python utils.py --root_img /path/to/images/dir ----root_ann /path/to/annotatations/dir

bash train.sh

bash predict.sh

Name		Name	Last commit message	Last commit date
Latest commit History 13 Commits
data		data
model		model
pictures		pictures
utils		utils
.gitignore		.gitignore
README.md		README.md
__init__.py		__init__.py
generate_policy_dataset.py		generate_policy_dataset.py
generate_policy_dataset.sh		generate_policy_dataset.sh
generating_video.py		generating_video.py
generating_video.sh		generating_video.sh
generating_video_segment.py		generating_video_segment.py
generating_video_segment.sh		generating_video_segment.sh
losses.png		losses.png
output.mp4		output.mp4
plotting_trajectory.py		plotting_trajectory.py
plotting_trajectory.sh		plotting_trajectory.sh
plotting_trajectory_segment.py		plotting_trajectory_segment.py
plotting_trajectory_segment.sh		plotting_trajectory_segment.sh
predict.py		predict.py
predict.sh		predict.sh
predict_ood.py		predict_ood.py
predict_ood.sh		predict_ood.sh
train.py		train.py
train.sh		train.sh