Code Monkey home page Code Monkey logo

wbia-tpl-2d-orientation's Introduction

2D-Orientation-v2

Refactor and improvements from the original 2D orientation project

python train.py -h

The program runs as follows:

if(example or save-all-figs):
  do example
elif(test):
  do test
else:
  train

Option Descriptions

--type: 
used to specifify a unique name for different optimizations that arent uncluded in  
progress folder naming, otherwize it would write over a state dict that has different  
optimization parameters but the same name
  options used to name progress folder:
    animal
    type
    nClasses
    pretrain
--no-resume: 
clean state dict and start training from scratch
--pretrain: 
use pretrained weights from official pytorch densenet
--separate-trig: 
instead of estimating the angle theta, the network has two outputs corresponding to  
cos and sin of the angle theta, and uses arctan2 to obtain theta. This is important  
because it changes nClasses to 2 instead of 1. 
--degree loss: 
only useful when determining if estimating the angle in degrees or radians increases  
accuracy

Example Usage

Show an Example of Current Estimation:

python3 train.py --type regression --nClasses 2 --device 0 --separate-trig --batchSz 3 --animal seadragon --example

Training (using slurm):

srun --time=240 --gres=gpu:1 --ntasks=1 python3 train.py --type regression --nClasses 2 --device 0 --separate-trig --batchSz 50 --animal seadragon

Show Loss History:

python3 train.py --type regression --nClasses 2 --separate-trig --pretrain --plot-loss-history --animal seadragon

wbia-tpl-2d-orientation's People

Contributors

hgrov52 avatar bluemellophone avatar

Watchers

James Cloos avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.