grasp_train.py integrate pixel-wise loss option

jhu-lcsr / costar_plan

Integrating learning and task planning for robots with Keras, including simulation, real robot, and multiple dataset support.

https://sites.google.com/site/costardataset

Apache License 2.0

68 stars 23 forks source link

grasp_train.py integrate pixel-wise loss option #366

Open ahundt opened 6 years ago

ahundt commented 6 years ago

We need to integrate pixel-wise loss into grasp_train.py, implementations are available in keras-contrib but we also need to account for the coordinate of the gripper.

[x] pixel wise loss at gripper coordinate
[x] gaussian loss at gripper coordinate
[x] ensure standard loss functions work too
- [x] add configuration of single pixel to grasp_train
- [ ] add configuration of gaussian to grasp_train
- [ ] add unit test for both

ahundt commented 6 years ago

see also #415 gaussian loss crash during training