How to use a ready-trained neural network?

MACHOCAPTCHA · November 23, 2021, 7:09pm

I found a neural network model that gets a depth map from an image in GitHub - OniroAI/MonoDepth-PyTorch: Unofficial implementation of Unsupervised Monocular Depth Estimation neural network MonoDepth in PyTorch. monodepth_resnet18_001.pth. How can I use this network? How can I just test it in my Pytorch?

ptrblck · November 23, 2021, 8:55pm

You could read through the linked repository and see how the model is initialized and used. I would probably start by looking for a “main” or “train” function first. Once this is done, you could then try to rip out the model and use it in your own custom script.

Eta_C · November 24, 2021, 2:04am

BTW, DONOT forget torch.no_grad() and model.eval()

MACHOCAPTCHA · December 12, 2021, 7:32am

Thank you for your reply! I was busy for a while and could not continue my studies in this direction. Now I am here and would like to ask if there is already a more detailed discussion or article. Which says how can I use a ready-made neural network step by step?

MACHOCAPTCHA · December 12, 2021, 7:33am

Unfortunately, so far I don’t know what it says.

my3bikaht · December 12, 2021, 9:28am

Come on, there is even inference notebook attached

github.com

OniroAI/MonoDepth-PyTorch/blob/master/Monodepth.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import torch\n",
    "import numpy as np\n",
    "import skimage.transform\n",
    "import matplotlib.pyplot as plt\n",
    "from easydict import EasyDict as edict\n",
    "\n",
    "from main_monodepth_pytorch import Model\n",
    "%reload_ext autoreload\n",
    "%autoreload 2"
   ]
  },

This file has been truncated. show original

MACHOCAPTCHA · December 12, 2021, 2:23pm

I have an error with these lines, I don’t know if I need them. The error sounds like: unexpected token ‘%’

my3bikaht · December 12, 2021, 7:45pm

These are commands for IPython (if you’re running script using Jupyter)

MACHOCAPTCHA · December 13, 2021, 8:07pm

thank you! I got it.