The ICEBERG Penguin colony usecase package
Project description
Quality Metrics
Prerequisites
- Linux
- Python 3
- CPU or NVIDIA GPU + CUDA CuDNN
Software Dependencies
- scipy==1.2.1
- Pillow==4.3.0
- torch
- scikit-learn==0.19.1
- torchvision==0.2.0'
- opencv-python
- rasterio
- future
Installation
PSC Bridges
From source:
$ git clone https://github.com/iceberg-project/Penguins.git
$ module load cuda
$ module load python3
$ virtualenv iceberg_penguins
$ source iceberg_penguins/bin/activate
[iceberg_penguins] $ export PYTHONPATH=<path>/iceberg_penguins/lib/python3.5/site-packages
[iceberg_penguins] $ pip install . --upgrade
From PyPi:
$ module load cuda
$ module load python3
$ virtualenv iceberg_penguins
$ source iceberg_penguins/bin/activate
[iceberg_penguins] $ export PYTHONPATH=<path>/iceberg_penguins/lib/python3.5/site-packages
[iceberg_penguins] $ pip install iceberg_penguins.search
To test
[iceberg_penguins] $ iceberg_penguins.detect
Prediction
- Download a pre-trained model at:
https://drive.google.com/file/d/149j5rlynkO1jQTLOMpL5lextHY0ozw6N/view?usp=sharing
Please put the model file to: <checkpoints_dir>/<model_name>/
The one provided here is at the epoch 300 of the model named "v3weakly_unetr_bs96_main_model_ignore_bad"
- The script to run the testing for a single PNG image:
iceberg_penguins.detect [--params ...]
iceberg_penguins.detect --gpu-ids 0 --name v3weakly_unetr_bs96_main_model_ignore_bad --epoch 300 --checkpoints_dir '../checkpoints_CVPR19W/' --output test --input_im ../data/Penguins/Test/A/GE01_20120308222215_1050410000422100_12MAR08222215-M1BS-054072905140_01_P002_u08rf3031.png
params:
- --name: name of the model used for testing
- --gpu_ids: the gpu used for testing
- --checkpoints_dir: path to the folder containing the trained models
- --epoch: which epoch we use to test the model
- --input_im: path to the input image
- --output: directory to save the outputs
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Hashes for iceberg_penguins.search-0.2.8.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 318c88dcd790f2fad9b373176e6ef5d0bede90218f661800e37ab3606ccfc13b |
|
MD5 | 7689f9ecf0feda2b0a794919ff169dcf |
|
BLAKE2b-256 | cdcd26b19a236f75fc3f78a63f20e8189a6b4a2fdc506d95d4123bbeb2d86650 |