JAEarly / bonfire

A backend library for multiple instance learning implemented in PyTorch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Bonfire

A backend library for multiple instance learning implemented in PyTorch.

Disclaimer
This is a WIP personal project, so there will often be changes to the main branch and backward compatibility is not guaranteed.

Installation

  1. Clone in as submodule:
git submodule add git@github.com:JAEarly/bonfire.git
  1. Install submodule as an editable package:
pip install -e bonfire
  1. Manually install PyTorch with CUDA support.

About

A backend library for multiple instance learning implemented in PyTorch.


Languages

Language:Python 100.0%