escorciav / roi_pooling

ROIPooling for pytorch
51 stars 13 forks source link

Add colab-setup to README #5

Closed escorciav closed 5 years ago

escorciav commented 5 years ago

Showcase the operation with a colab notebook.

This may not run because of #4 . In case you are eager to see it in action, a quick-fix would be:

  1. updating roi_pooling.functions.roi_pooling.py by replacing Dtype(argmax_data) by int on the lines mentioned in #4 .
  2. remove the old file in colab
  3. Add the file with the update mentioned in 1.
  4. enjoy 🍻