mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
@crossbot I will murder you in your sleep ⚔️
This commit is contained in:
71
examples/art.c
Normal file
71
examples/art.c
Normal file
@ -0,0 +1,71 @@
|
||||
#include "darknet/network.h"
|
||||
#include "darknet/utils.h"
|
||||
#include "darknet/parser.h"
|
||||
#include "darknet/option_list.h"
|
||||
#include "darknet/blas.h"
|
||||
#include "darknet/classifier.h"
|
||||
#include <sys/time.h>
|
||||
|
||||
void demo_art(char *cfgfile, char *weightfile, int cam_index)
|
||||
{
|
||||
#ifdef OPENCV
|
||||
network net = parse_network_cfg(cfgfile);
|
||||
if(weightfile){
|
||||
load_weights(&net, weightfile);
|
||||
}
|
||||
set_batch_network(&net, 1);
|
||||
|
||||
srand(2222222);
|
||||
CvCapture * cap;
|
||||
|
||||
cap = cvCaptureFromCAM(cam_index);
|
||||
|
||||
char *window = "ArtJudgementBot9000!!!";
|
||||
if(!cap) error("Couldn't connect to webcam.\n");
|
||||
cvNamedWindow(window, CV_WINDOW_NORMAL);
|
||||
cvResizeWindow(window, 512, 512);
|
||||
int i;
|
||||
int idx[] = {37, 401, 434};
|
||||
int n = sizeof(idx)/sizeof(idx[0]);
|
||||
|
||||
while(1){
|
||||
image in = get_image_from_stream(cap);
|
||||
image in_s = resize_image(in, net.w, net.h);
|
||||
show_image(in, window);
|
||||
|
||||
float *p = network_predict(net, in_s.data);
|
||||
|
||||
printf("\033[2J");
|
||||
printf("\033[1;1H");
|
||||
|
||||
float score = 0;
|
||||
for(i = 0; i < n; ++i){
|
||||
float s = p[idx[i]];
|
||||
if (s > score) score = s;
|
||||
}
|
||||
score = score;
|
||||
printf("I APPRECIATE THIS ARTWORK: %10.7f%%\n", score*100);
|
||||
printf("[");
|
||||
int upper = 30;
|
||||
for(i = 0; i < upper; ++i){
|
||||
printf("%c", ((i+.5) < score*upper) ? 219 : ' ');
|
||||
}
|
||||
printf("]\n");
|
||||
|
||||
free_image(in_s);
|
||||
free_image(in);
|
||||
|
||||
cvWaitKey(1);
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
void run_art(int argc, char **argv)
|
||||
{
|
||||
int cam_index = find_int_arg(argc, argv, "-c", 0);
|
||||
char *cfg = argv[2];
|
||||
char *weights = argv[3];
|
||||
demo_art(cfg, weights, cam_index);
|
||||
}
|
||||
|
Reference in New Issue
Block a user