Skip to content

Commit

Permalink
drivefileid helper funcs
Browse files Browse the repository at this point in the history
  • Loading branch information
aubricot committed Jan 17, 2025
1 parent cf51798 commit f165e44
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,8 @@
"\n",
"# Install dependencies\n",
"!pip3 install --upgrade gdown\n",
"!gdown 1fIEf387CNrWk0ziPY-ltvwN9VrRXrRkY # Download helper_funcs folder\n",
"!tar -xzvf helper_funcs.tar.gz -C .\n",
"!gdown 1-IEwduCmOWHc5uk3oKQsxeHBihS7wZtl # Download helper_funcs folder\n",
"!tar -xzvf aves_helper_funcs.tar.gz -C .\n",
"!pip install -r requirements.txt"
]
},
Expand All @@ -91,7 +91,7 @@
"%cd $cwd\n",
"\n",
"# Load Pre-trained model from Tensorflow Hub (both trained on MS COCO 2017)\n",
"model = \"SSD MobileNet v2\" #@param [\"SSD MobileNet v2\", \"Faster RCNN Resnet 50\"] {allow-input: true}\n",
"model = \"Faster RCNN Resnet 50\" #@param [\"SSD MobileNet v2\", \"Faster RCNN Resnet 50\"] {allow-input: true}\n",
"detector, module_handle, mod_abbv = load_tfhub_detector(model)\n",
"\n",
"# Load corresponding label map for MS COCO 2017\n",
Expand Down Expand Up @@ -258,7 +258,7 @@
"\n",
"# Take 5k subset of bundle for running inference\n",
"# Change filename for each batch\n",
"crops_file = \"aves_cropcoords_tf2_c\" #@param [\"aves_cropcoords_tf2_a\", \"aves_cropcoords_tf2_b\", \"aves_cropcoords_tf2_c\", \"aves_cropcoords_tf2_d\"] {allow-input: true}\n",
"crops_file = \"aves_cropcoords_tf2_b\" #@param [\"aves_cropcoords_tf2_a\", \"aves_cropcoords_tf2_b\", \"aves_cropcoords_tf2_c\", \"aves_cropcoords_tf2_d\"] {allow-input: true}\n",
"outfpath = set_outpath(crops_file, cwd)\n",
"\n",
"# Write header row of output tag file\n",
Expand Down

0 comments on commit f165e44

Please sign in to comment.