Skip to content

HaonanPeng/active_syn_generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 

Repository files navigation

Active learning with generated synthetic image for instrument segmentation

Important update on May.24 2024: In 2024, Google Colab no longer supports TensorFlow 1.13.1 which is required by our system. The training cannot be performed on Colab now. Please refer to the new local training version. Please notice that this is the ResNet version which takes longer training time than the previous MobileNet version. And manual setup of the workspace is necessary. Please contact Haonan Peng (penghn@uw.edu) if you need help and I will try to respond timely. CUDA 10 and VS 2017 are necessary in addition to the Python 3.6 environment.

The 3 '.ipynb' script can be run in Google Colab. To run it on your Colab, paths in the following cells should be modified to fit your Google Drive path.

import sys
path_workspace = '/content/active_learning_v1'
path_source = '/content/drive/MyDrive/active_learning_ws/active_learning_v1'
sys.path.append(path_source)

Source files for UW Sinus datasets can be found in 'src/active_syn_UW_sinus'

Source files for Endovis datasets can be found in 'src/active_syn_Endovis'

import shutil
!mkdir '/content/active_learning_v1'

shutil.copy("/content/drive/MyDrive/active_learning_ws/active_learning_v1/labels.json", 
            "/content/active_learning_v1/labels.json")

!unzip -q "/content/drive/MyDrive/active_learning_ws/data/database_folder12_l.zip" -d '/content/active_learning_v1'
!unzip -q "/content/drive/MyDrive/active_learning_ws/data/background_for_syn_folder12_l.zip" -d '/content/active_learning_v1'
!unzip -q "/content/drive/MyDrive/active_learning_ws/active_learning_v1/mobilenet_v1_1.0_224.zip" -d '/content/active_learning_v1'

Link to UW Sinus Live/Cadaver dataset

Link to EndoVis 2017 dataset

Link to mobilenet_v1_1.0_224

Link to ResNet50 V2

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published