Skip to content

Commit

Permalink
Minor fix
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexeyAB committed Oct 13, 2018
1 parent 344f8c6 commit 2bb4445
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 1 deletion.
3 changes: 3 additions & 0 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,9 @@ OS := $(shell uname)
# Tesla V100
# ARCH= -gencode arch=compute_70,code=[sm_70,compute_70]

# GeForce RTX 2080 Ti, RTX 2080, RTX 2070 Quadro RTX 8000, Quadro RTX 6000, Quadro RTX 5000 Tesla T4
# ARCH= -gencode arch=compute_75,code=[sm_75,compute_75]

# Jetson XAVIER
# ARCH= -gencode arch=compute_72,code=[sm_72,compute_72]

Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -370,7 +370,7 @@ If you made you custom model that isn't based on other models, then you can trai

## When should I stop training:

Usually sufficient 2000 iterations for each class(object). But for a more precise definition when you should stop training, use the following manual:
Usually sufficient 2000 iterations for each class(object), but not less than 4000 iterations in total. But for a more precise definition when you should stop training, use the following manual:

1. During training, you will see varying indicators of error, and you should stop when no longer decreases **0.XXXXXXX avg**:

Expand Down
5 changes: 5 additions & 0 deletions src/http_stream.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -322,6 +322,11 @@ image image_data_augmentation(IplImage* ipl, int w, int h,
sized *= dexp;
}

//std::stringstream window_name;
//window_name << "augmentation - " << ipl;
//cv::imshow(window_name.str(), sized);
//cv::waitKey(0);

// Mat -> IplImage -> image
IplImage src = sized;
image out = ipl_to_image(&src);
Expand Down

0 comments on commit 2bb4445

Please sign in to comment.